zoomy_jax

zoomy_jax — JAX backend for Zoomy.

Sets the process-global JAX float precision (x64 by default) and the persistent XLA compilation cache at package import, since both must be configured before the first array or compile.

Subpackages

• zoomy_jax.symsolver

Submodules

• zoomy_jax.gpurank

Package Contents

Data

ZOOMY_JAX_ENABLE_X64

API

zoomy_jax.ZOOMY_JAX_ENABLE_X64

‘_env_on(…)’