Skip to main content
Ctrl+K

JAX internals#

Note

This is a placeholder for a section in the new JAX tutorials.

For the time being, you may find some related content in the old documentation:

  • How JAX primitives work

  • Understanding Jaxprs

  • Writing custom Jaxpr interpreters in JAX

  • Type promotion semantics