JAX Internal Implementation Notes
JAX Internal Implementation Notes#
This section of the documentation describes implementation details thatare too elaborate to fit in code comments, or in internal function docstrings.
Like code comments, the material here should be used with caution because itis prone to become stale.
