AI Tools / AI products & services
jax.export is an API in the JAX Python library for exporting staged computations as serialized, StableHLO-based programs. It supports ahead-of-time lowering and shape-polymorphic exports for execution in other environments.