Flax
Neural network library for JAX by Google providing Linen module system.
Open SourceSelf HostedOffline Capable
0.0 (0)
About
Flax by Google is a neural network library for JAX. Provides the Linen module system for defining models, training state management, and serialization. Used in many Google research projects. Apache 2.0 license.
Reviews (0)
Leave a Review
No reviews yet. Be the first to review!
Details
- Category
- AI Frameworks & Libraries
- Price
- Free
- Platform
- Local/Desktop
- Difficulty
- Advanced (4/5)
- License
- Apache-2.0
- Added
- Apr 3, 2026
Similar Tools
Featured
Open-source machine learning framework by Meta with dynamic computation graphs.
Open SourceSelf HostedOffline
Intermediate
0.0 (0)
Featured
End-to-end open-source ML platform by Google for training and deployment.
Open SourceSelf HostedOffline
Intermediate
0.0 (0)
High-performance numerical computing library by Google with auto-differentiation.
Open SourceSelf HostedOffline
Advanced
0.0 (0)