jax.enable_x64

jax.enable_x64 = <jax._src.config.State object>

Context manager for jax_enable_x64 config option.

Enable 64-bit types to be used

Parameters:

new_val (Any)