AI & ML ecosystem
Deep learning
JAX
Composable transforms (grad/jit/vmap) over NumPy; favoured for research at scale.
- Role
- Deep learning
- Maintainer
- Google DeepMind
- Languages
- Python
- Stars
- 36K
- Forks
- 3.7K
- Open issues
- 2.6K
- Last updated
- today
Links
Related entities
Maintained by
Related model families