It's built on JAX, not vanilla Python. The metric being optimized is not just performance, but also the ability to build reasonably performant workflows with a

It's built on JAX, not vanilla Python. The metric being optimized is not just performance, but also the ability to build reasonably performant workflows with arbitrary differentiable (i.e., ML) inputs and outputs.