> JAX is Autograd and XLA, brought together for high-performance machine learning research.
That does not really convey the generality of it that well.
> JAX is Autograd and XLA, brought together for high-performance machine learning research.
That does not really convey the generality of it that well.