vhZ HdZddlmZddlmZeddefdZedZy) zeContext managers for toggling X64 mode. **Experimental: please give feedback, and expect changes.** )contextmanager)confignew_valc#hKtj|5ddddy#1swYyxYww)auExperimental context manager to temporarily enable X64 mode. .. warning:: This context manager remains experimental because it is fundamentally broken and can result in unexpected behavior, particularly when used in conjunction with JAX transformations like :func:`jax.jit`, :func:`jax.vmap`, :func:`jax.grad`, and others. See https://github.com/jax-ml/jax/issues/5982 for details. Usage:: >>> x = np.arange(5, dtype='float64') >>> with enable_x64(): ... print(jnp.asarray(x).dtype) ... float64 See Also -------- jax.experimental.disable_x64 : temporarily disable X64 mode. Nr enable_x64)rs W/opt/face_recognition/venv/lib/python3.12/site-packages/jax/experimental/x64_context.pyrrs-.!     2& 2/2c#hKtjd5ddddy#1swYyxYww)auExperimental context manager to temporarily disable X64 mode. .. warning:: This context manager remains experimental because it is fundamentally broken and can result in unexpected behavior, particularly when used in conjunction with JAX transformations like :func:`jax.jit`, :func:`jax.vmap`, :func:`jax.grad`, and others. See https://github.com/jax-ml/jax/issues/5982 for details. 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_x64r7s-.     r N)T)__doc__ contextlibrjax._srcrboolrrr r r rs?&   2  r