uki dZddlmZddlmZed defdZedZdefdefd Zdd l Z e jreZ eZ [ y dd lmZeeeZ[[ y ) z]Context managers for toggling X64 mode. **Deprecated: use :func:`jax.enable_x64` instead.** )contextmanager)confignew_valc#hKtj|5ddddy#1swYyxYww)aExperimental context manager to temporarily enable X64 mode. .. warning:: This context manager is deprecated as of JAX v0.8.0, and will be removed in JAX v0.9.0. Use :func:`jax.enable_x64` instead. Usage:: >>> import jax >>> x = np.arange(5, dtype='float64') >>> with _enable_x64(True): ... print(jnp.asarray(x).dtype) ... float64 See Also -------- jax.experimental.disable_x64 : temporarily disable X64 mode. Nr enable_x64)rs W/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/experimental/x64_context.py _enable_x64r s-,!     2& 2/2c#hKtjd5ddddy#1swYyxYww)aExperimental context manager to temporarily disable X64 mode. .. warning:: This context manager is deprecated as of JAX v0.8.0, and will be removed in JAX v0.9.0. Use :func:`jax.enable_x64` instead. Usage:: >>> x = np.arange(5, dtype='float64') >>> with _disable_x64(): ... print(jnp.asarray(x).dtype) ... float32 See Also -------- jax.experimental.enable_x64 : temporarily enable X64 mode. FNrr _disable_x64r0s-*     r zjax.experimental.x64_context.disable_x64 is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.enable_x64(False) instead.zjax.experimental.x64_context.enable_x64 is deprecated in JAX v0.8.0 and will be removed in JAX v0.9.0; use jax.enable_x64(True) instead.) disable_x64rN)deprecation_getattr)T)__doc__ contextlibrjax._srcrboolr r _deprecationstyping_typing TYPE_CHECKINGrrjax._src.deprecationsr_deprecation_getattr__name__ __getattr__r rr rs &   0  4: 9   *+  P$X}=+ r