Ë
    ñuki–  ã                   ó    — d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZ d dlmZmZ ddiZd dlZej&                  r	 [yd dlmZ  eee«      Z[[y)é    )Ú_initial_style_jaxprÚ_sum_tangentsÚ_zeros_like_pytreeÚclosure_convertÚcustom_gradientÚ
custom_jvpÚcustom_jvp_call_pÚ
custom_vjpÚcustom_vjp_call_pÚcustom_vjp_primal_tree_valuesÚCustomVJPPrimalÚlinear_callÚremat_opt_p)ÚSymbolicZeroÚzero_from_primalÚcustom_jvp_call_jaxpr_p)zÖjax.custom_derivatives.custom_jvp_call_jaxpr_p was deprecated in v0.7.0 and removed in v0.8.0. use jax.extend.core.primitives.custom_jvp_call_p instead, and please note that you must `import jax.extend` explicitly.NN)Údeprecation_getattr)Újax._src.custom_derivativesr   r   r   r   r   r   r	   r
   r   r   r   r   r   Újax._src.ad_utilr   r   Ú_deprecationsÚtypingÚTYPE_CHECKINGÚjax._src.deprecationsr   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úQ/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/custom_derivatives.pyú<module>r       s]   ð÷$÷ ÷ õ ÷ ð ð  ð€ó Ø	×ÒØñ
 õ PÙ$ X¨}Ó=€+ØÙ
r   