rocm_jax/docs/tutorials/jax-internals.md

JAX internals

This is a placeholder for a section in the new {ref}`jax-tutorials`.

For the time being, you may find some related content in the old documentation:
- {doc}`../notebooks/How_JAX_primitives_work`
- {doc}`../jaxpr`
- {doc}`../notebooks/Writing_custom_interpreters_in_Jax`
- {doc}`../type_promotion`