jax.experimental module — JAX documentation (original) (raw)

By The JAX authors

© Copyright 2024, The JAX Authors.