"JAX is widely used in machine learning and scientific computing, the latter of which often relies on existing high-performance code that we would ideally like to incorporate into JAX."
"Connecting code to JAX requires knowledge of the internals of JAX and its C++ backend."