google/jax
jax, JAX, на сайте с December 16, 2022 14:03
JAX is Autograd and XLA, brought together for high-performance machine learning research.