
    uki)
                     h   d dl mZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ dej                  fdej                  fd	ej                   fd
ej"                  fdZd dlZej*                  r4ej                   Zej"                  Zej                  Zej                  Z[[[yd dlmZ  eee      Z[[[[y)    )io_callback)primal_tangent_dtype)EArray)core)cur_qdd)x64_contextz~jax.experimental.disable_x64 is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.enable_x64(False) instead.z|jax.experimental.enable_x64 is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.enable_x64(True) instead.zvjax.experimental.mutable_array is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.new_ref instead.zqjax.experimental.MutableArray is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.Ref instead.)disable_x64
enable_x64mutable_arrayMutableArrayN)deprecation_getattr)jax._src.callbackr   jax._src.dtypesr   jax._src.earrayr   jax._srcr   	_src_corejax._src.corer   jax.experimentalr   _x64_context_disable_x64_enable_x64new_refRef_deprecationstyping_typingTYPE_CHECKINGr   r   r
   r	   jax._src.deprecationsr   _deprecation_getattr__name____getattr__     T/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/experimental/__init__.py<module>r%      s   0 ' 9
:90,MM#0 
##-,''*))+
  P$X}=+r#   