Jax

Autograd and XLA for high-performance machine learning research.

GitHub repo Directory import
0 shouts Added Sep 23, 2026 github.com
github.com/google/jax
Jax website preview

Links

Tags