Internal APIs ============= core ---- .. currentmodule:: jax.core .. automodule:: jax.core .. autosummary:: :toctree: _autosummary Jaxpr ClosedJaxpr