
    uki7                         d dl mZ eZd dlmZ d dlmZ ej                  j                  Z[ddiZd dlZej                  r	 [yd dlmZ  eee      Z[[y)    )canonicalize_value_handlers)apply_primitive)
xla_clientcanonicalize_dtype)zjax.interpreters.xla.canonicalize_dtype was deprecated in JAX v0.7.0 and removed in JAX v0.8.0. For canonicalizing dtypes, prefer jax.dtypes.canonicalize_dtype. For checking whether an object is a valid jax input, prefer jax.core.valid_jaxtype.NN)deprecation_getattr)jax._src.dtypesr   _canonicalize_value_handlerscanonicalize_dtype_handlersjax._src.dispatchr   jax._src.libr   _xc_xlaClientBackend_deprecationstyping_typingTYPE_CHECKINGjax._src.deprecationsr   _deprecation_getattr__name____getattr__     O/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/interpreters/xla.py<module>r      sk    ;  +
((//
   

  P$X}=+r   