No known security advisories.
The current version has no OSV.dev malware reports and no critical or high-severity vulnerabilities.
flax has 5.1M monthly downloads and 7K GitHub stars. Version 0.12.7 supports Python 3.11+, first published in 2020.
It is actively maintained under a permissive license, last released in Apr 2026.
It is built on top of jax, msgpack and numpy. It is used by 366 tracked packages, including tensorflowjs, clu and jax-md.
GitHub topics include jax.
Package Insights
Score breakdown - each component scores 0-100, weighted by percentage
Maintenance
100
Popularity
83
Adoption
60
Community
80
Provenance
100
How the maintenance status was determined
Last commit
4 days ago
Monthly downloads
5,085,841
•
Active
Activity within 30 days
Stable
Activity within 31-180 days
Minimal
Activity within 181-365 days, or no activity data
At Risk
No activity for > 365 days, but > 10,000 monthly downloads
Archived
Repository flagged as archived on GitHub
How the popularity category was determined
Monthly downloads
5,085,841
Stars
7,254
Stars per 1k downloads
1.43
Industry Standard
> 10,000,000 monthly downloads
Hidden Gem
< 0.5 stars per 1k downloads and > 100,000 monthly downloads
Community Favorite
> 5 stars per 1k downloads
•
Rising Star
Balanced ratio of stars to downloads
Unknown
Fewer than 1,000 monthly downloads, or no star data
How download momentum is calculated
Week daily avg
147,015
Month daily avg
169,528
Momentum
-13.3%
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Python version support tiers
requires_python
>=3.11
Cutting Edge
Minimum supported version is 3.12 or newer
•
Modern
Minimum supported version is 3.9, 3.10, or 3.11
Legacy
Supports 3.8 or older (or Python 2)
Unknown
No requires_python specified or unparseable
License categories
License
Apache Software License
•
Permissive
MIT, Apache, BSD, ISC, Unlicense, CC0, etc.
Copyleft
GPL, LGPL, AGPL, MPL, EPL, etc.
Proprietary
Commercial, all rights reserved, custom EULAs
Unknown
No license specified, or none of the above match
How the release cadence was determined
Last release
75 days ago
Total releases
64
Releases/year
10.1
Rapid
Last release within 365 days and > 12 releases/year
•
Regular
Last release within 365 days
Slow
Last release 366-730 days ago
Stalled
Last release > 730 days ago, or no releases
Unknown
No usable release date available
How contributor risk was determined
Contributors
297
Monthly downloads
5,085,841
High Load
≤ 2 contributors and ≥ 50,000 monthly downloads
Moderate Load
< 10 contributors and ≥ 500,000 monthly downloads
Low Load
< 25 contributors and ≥ 5,000,000 monthly downloads, or sole maintainer with ≥ 10,000 monthly downloads
•
Balanced
Enough contributors for the package's popularity
Unknown
No contributor data available
Engagement
86
Issue Burden
31.4
per 1k stars
Discussions
106.3
per 1k stars
Fork Ratio
36.2%
contrib / forks
Stale Release
71d
PR Velocity
Active
Weekly Downloads
GitHub Stars
Not all downloads include detailed metadata. Breakdowns reflect the subset that does.
Downloads by OS
Linux
92.0K
macOS
9.1K
Windows
5.9K
Python Versions
3.12
62.5%
65.7K
3.11
23.3%
24.4K
3.10
8.8%
9.2K
3.13
3.9%
4.1K
3.8
1.6%
1.6K
Top Countries
1.
US
96.1K
US
2.
SG
2.4K
SG
3.
NL
1.3K
NL
4.
SE
1.1K
SE
5.
IE
955
IE
6.
GB
710
GB
7.
HK
652
HK
8.
JP
652
JP
9.
CN
605
CN
10.
MY
455
MY
Dependencies
10
- jax >=0.10.0
- msgpack
- numpy >=1.23.2
- optax
- orbax-checkpoint
- pyyaml >=5.4.1
- rich >=11.1
- tensorstore
- treescope >=0.1.7
- typing-extensions >=4.2
48 optional dependencies
- ale-py[testing]
- cloudpickle[testing]
- clu[testing]
- dm-haiku[docs]
- docutils[docs]
- einops[testing]
- einops[docs]
- grain[testing]
- gymnasium[testing]
- ipykernel[docs]
- ipython-genutils[docs]
- ipywidgets[docs]
- jaxlib[testing]
- jaxtyping[testing]
- jraph[testing]
- jupytext[docs]
- kagglehub[docs]
- keras[testing]
- matplotlib[docs]
- ml-collections[docs]
- ml-collections[testing]
- mypy[testing]
- myst-nb[docs]
- nanobind[dev]
- nbstripout[docs]
- opencv-python[testing]
- orbax-export[docs]
- pre-commit[dev]
- protobuf[testing]
- pygments[docs]
- pytest[testing]
- pytest-cov[testing]
- pytest-custom-exit-code[testing]
- pytest-xdist[testing]
- pytype[testing]
- recommonmark[docs]
- scikit-build-core[dev]
- scikit-learn[docs]
- sentencepiece[testing]
- sphinx[docs]
- sphinx-book-theme[docs]
- sphinx-design[docs]
- tensorflow[testing]
- tensorflow-datasets[testing]
- tensorflow-text[testing]
- torch[testing]
- tqdm[docs]
- treescope[testing]
Used By
366
- tensorflowjs
- clu
- jax-md
- qwix
- aqtp
- google-tunix
- tpu-inference
- dopamine-rl
- gymnax
- brax
- google-metrax
- kauldron
- gemma
- jax2onnx
- tojax
- fusion-surrogates
- evosax
- jax-ai-stack
- praxis
- roxxel
- netket
- playground
- ximinf
- paxml
- jax-tpu-embedding
- mplang-nightly
- craftax
- pertpy
- flashbax
- alberta-framework
- transformer-engine-jax
- jaxmarl
- sbx-rl
- hyperbolix
- pycse
- masa-safe-rl
- navix
- torax
- easydel
- mlip
- jax-image-models
- moscot
- sinapsis-huggingface-transformers
- pyrddlgym-jax
- qdax
- cd-dynamax
- mintstate
- cax
- crazyflow
- fjformer
Package Information
License
Apache Software License
First Release
March 17, 2020
Last Release
2 months ago
Last Commit
5 days ago
Latest Version
0.12.7
Total Releases
64
Contributors
297
Open PRs
264
Archived
No
Python Versions
3.11
3.12
Wheels
py3
Tracking Since
January 6, 2026
Data Updated
July 5, 2026
Data Freshness
Fresh