- Category
- AI Tools
- Pricing
- Open Source
- Type
- TOOL
- Builder
- GitHub
- 7.3k stars
- Added
- May 22, 2026
About
Google's neural network library for JAX — flexible by design, used by Gemini, AlphaFold, and most JAX-based research projects.
Why it made the leaderboard
Google's neural network library for JAX, battle-tested on Gemini and AlphaFold — if you're building JAX models, this is the layer API most of the research ecosystem already speaks.
Tags
jaxneural-networksgooglemllibrary
Tech Stack
Python
Comments (0)
No comments yet
Indexed by a proprietary survey. Corrections welcome.