Training & fine-tuning
Flax
Adds expressive neural-network structure and state management to JAX’s program-transformation model.
Overview
Research summary
Flax provides neural-network abstractions on top of JAX. Its NNX interface uses regular Python objects and reference semantics to make models easier to construct, inspect and modify, while the earlier Linen interface remains separately documented. Developers combine Flax models with JAX transformations and optimization libraries to implement training workflows. The library emphasizes flexibility for machine-learning research and experimentation. It supplies model-building components rather than a hosted training platform or a collection of universally licensed pretrained weights.
Repository summary
- Stars
- 7,334
- Open issues
- Unavailable
- Last push
- 2026-10-01
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Version
- Unavailable
Recorded catalogue figures. View repository data and provenance →
Classification
Pricing & services
Paid services unknown
Whether the provider offers paid products or services has not been established.
Licence scope
Implementation
Recorded implementation details and interfaces for Flax.
Implementation details
Python library and supporting tools
- Languages
- Python
- Repository type
- source
Recorded interfaces and capabilities
Licence scope
Repository
Repository snapshots, release information and recorded maintenance signals.
Repository snapshot
- Stars
- 7,334
- Open issues
- Unavailable
- Last push
- 2026-10-01
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Archived
- Not recorded
Repository activity is a snapshot, not a quality or popularity ranking. It combines recent-push freshness (50%), 90-day commits (30%) and issue pressure (20%).
Maintenance and provenance
- Catalogue snapshot
- 2026-10-06
- Stars source
- Recorded fallback
- Last push source
- Recorded fallback
Documentation
Recorded references and research provenance for this entry.
Recorded sources 5
- https://flax.readthedocs.io Project page · Documentation
- https://github.com/google/flax Linked repository · Research reference
- https://api.github.com/repos/google/flax Research reference
- https://github.com/google/flax/blob/main/README.md Research reference
- https://github.com/google/flax/blob/main/LICENSE Research reference
Research metadata
- Research date
- 2026-10-02
- Catalogue snapshot
- 2026-10-06
Community & social 1 channels
Communities
- GitHub Discussionsgithub.com/google/flax/discussionsOfficial
Link details for GitHub Discussions
- Last checked
Related tools
Editorially linked entries to explore alongside Flax. This is not a ranking.
