jit

Auto-generated from jax_code_defs

PyTorch

API: torch.compile
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.jit
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.transforms.compilation.jit
Strategy: Direct Mapping
Official Docs ↗