google / flax
Flax is a neural network library for JAX that is designed for flexibility.
google/flax holds a health index of 95 out of 100, placing it in the Exceptional band. It scores highest on Vitality (99/100) and lowest on Security (21/100). It was last updated 2 days ago. 6 contributors account for most of its recent work.
Software health index
Metrics are grouped into weighted categories on one standardized 1–100 scale. Overall starts as their weighted mean, calibrated against the distribution of the public record so bands carry percentile meaning; when public evidence triggers the High-Risk Jurisdiction Policy, the rating is adjusted and receives an At Risk ceiling of 34.
Score profile
Each axis is a category. The shape matters more than the average — a healthy subject fills the whole shape, while a spike-and-crater profile means strength in one dimension is masking risk in another.
The weighted overall 82 is calibrated to 95 on the published index scale (record calibration 2026-08-02).
Ownership
This repository is backed by an organization — shared, accountable stewardship that can outlive any single maintainer.
Package ecosystems
| Registry | Package | Version | Downloads / mo | Versions | Last publish |
|---|---|---|---|---|---|
| PyPI | flax | 0.12.9 | 4,371,092 | 66 | 8 days ago |
| PyPI | flaxlib | 0.0.1a1 | 66 | 1 | 709 days ago |
Metrics by category
Vitality
Is the project alive — is code being written and are releases shipping?
Development activity
How it's scored
| 36/36 | Push recency — last push 2 days ago |
| 35.3/36 | Commit cadence — 51/52 weeks with commits |
| 18/18 | Commit volume — 639 commits in the last year |
| 0/10 | OpenSSF Scorecard: Maintained — no data |
Inputs used
| commits_last_year | 639 |
| human_commit_share | 1 |
| days_since_last_push | 2 |
| active_weeks_last_year | 51 |
Release discipline
How it's scored
| 27/27 | Ships releases — 64 releases published |
| 36/36 | Release recency — latest release 8 days ago |
| 27/27 | Release cadence — a release every ~36.3 days |
| 0/10 | OpenSSF Scorecard: Signed-Releases — no data |
Inputs used
| releases_count | 64 |
| latest_release_tag | v0.12.9 |
| releases_from_tags | no |
| days_since_latest_release | 8 |
| mean_days_between_releases | 36.3 |
Community & Adoption
Does the project have users, downloads, attention, and a welcoming setup for contributors?
Popularity & adoption
How it's scored
| 60/60 | Stars — 7,304 stars |
| 24.3/25 | Forks — 829 forks |
| 10.6/15 | Watchers — 81 watchers |
Inputs used
| forks | 829 |
| stars | 7,304 |
| watchers | 81 |
| growth_state | unverified |
| growth_factor_pct | 100 |
| growth_unverified_reason | no_history |
Community health
How it's scored
| 22.5/22.5 | README |
| 22.5/22.5 | License — recognized license (Apache-2.0) |
| 18/18 | CONTRIBUTING guide |
| 13.5/13.5 | Code of conduct |
| 0/7.2 | Issue template |
| 6.3/6.3 | PR template |
Inputs used
| has_readme | yes |
| has_license | yes |
| readme_badges | 2 |
| has_contributing | yes |
| has_issue_template | no |
| has_code_of_conduct | yes |
| readme_badge_services | github.com, shields.io |
| has_pull_request_template | yes |
Ecosystem adoption (downloads)
How it's scored
| 80/80 | Monthly downloads — 4,371,092 downloads/month across pypi |
| 0/20 | Registry dependents — not reported by this ecosystem |
Inputs used
| packages | flax |
| dependents | — |
| ecosystems | pypi |
| total_downloads | — |
| monthly_downloads | 4,371,092 |
| unverified_packages_excluded | flaxlib |
Sustainability & Governance
Will the project survive its people — bus factor, responsiveness, who backs it, and package upkeep?
Maintainer resilience (bus factor)
How it's scored
| 48.6/54 | Bus factor — 6 contributor(s) cover half of all commits |
| 19.6/22.5 | Commit distribution — top contributor authored 13% of commits |
| 13.5/13.5 | Contributor breadth — 99 contributors |
| 0/10 | OpenSSF Scorecard: Contributors — no data |
Inputs used
| bus_factor | 6 |
| contributors_sampled | 99 |
| top_contributor_share | 0.128 |
How it's scored
| 33.6/42 | Issue resolution — 80% of issues closed |
| 22.5/30 | PR acceptance — 2,424/3,231 decided PRs merged |
| 0/13 | Newcomer PR acceptance — 0/1 first-time contributors' PRs merged in 30d |
| 0/15 | OpenSSF Scorecard: Code-Review — no data |
Inputs used
| merged_prs | 2,424 |
| open_issues | 230 |
| closed_issues | 917 |
| prs_merged_7d | 0 |
| prs_decided_7d | 1 |
| prs_merged_30d | 2 |
| prs_decided_30d | 4 |
| issue_closed_ratio | 0.799 |
| closed_unmerged_prs | 807 |
| first_time_authors_30d | 1 |
| first_time_prs_merged_30d | 0 |
| first_time_prs_decided_30d | 1 |
Ownership & stewardship
How it's scored
| 30/30 | Ownership backing — organization-owned |
| 20/20 | Verified domain |
| 25/25 | Owner reach — 77,499 followers of google |
| 25/25 | Track record — 2,897 public repos, account ~14 yr old |
Inputs used
| followers | 77,499 |
| owner_type | Organization |
| is_verified | yes |
| owner_login | |
| public_repos | 2,897 |
| account_age_days | 5,335 |
Package maintenance
How it's scored
| 25/25 | Published & resolvable — 2 package(s) on pypi |
| 35/35 | Publish recency — latest publish 8 days ago |
| 20/20 | Version history — 66 published versions |
| 20/20 | Not deprecated — active, not deprecated or yanked |
Inputs used
| packages | flax, flaxlib |
| ecosystems | pypi |
| any_deprecated | no |
| min_days_since_publish | 8 |
Engineering Quality
Are baseline engineering and documentation practices in place?
Engineering practices
How it's scored
| 24/24 | CI workflows — 4 workflow(s) |
| 24/24 | Tests present |
| 16/16 | Linter config — pyproject.toml ([tool.ruff], [tool.yapf]) |
| 9.6/9.6 | Pre-commit hooks |
| 0/6.4 | .editorconfig |
| 0/20 | OpenSSF Scorecard: CI-Tests — no data |
Inputs used
| has_ci | yes |
| has_tests | yes |
| has_editorconfig | no |
| has_linter_config | yes |
| has_precommit_config | yes |
Documentation
How it's scored
| 30/30 | README |
| 25/25 | Documentation directory |
| 15/15 | Documentation / homepage site — https://flax.readthedocs.io |
| 10/10 | Repository description |
| 10/10 | Topics — 1 topics |
| 10/10 | Wiki |
Inputs used
| topics | jax |
| has_wiki | yes |
| homepage | https://flax.readthedocs.io |
| docs_site | https://flax.readthedocs.io |
| has_readme | yes |
| has_docs_dir | yes |
| has_description | yes |
Security
Are visible security and supply-chain practices strong, without unresolved high-risk jurisdiction exposure?
Security posture
How it's scored
| 0/30 | Security policy (SECURITY.md) |
| 0/25 | Dependabot config |
| 0/25 | Dependency lockfiles — published library — lockfiles are an application concern, not expected |
| 0/20 | CodeQL workflow |
Inputs used
| source | file_signals |
| lockfiles | Cargo.lock, uv.lock |
| manifests | flaxlib_src/Cargo.toml, flaxlib_src/pyproject.toml, pyproject.toml |
| has_codeql_workflow | no |
| has_security_policy | no |
| has_dependabot_config | no |
Dependency advisories
How it's scored
| 35/35 | Direct dependencies free of known advisories — no direct dependency carries a known advisory |
| 25/25 | Indirect dependencies free of known advisories — no indirect dependency carries a known advisory |
| 0/40 | No advisories left outstanding — no advisory carries a publication date |
Inputs used
| source | osv |
| advisories | 0 |
| affected_packages | 0 |
| assessed_packages | 28 |
| unassessed_packages | 0 |
| affected_by_severity | none |
| direct_affected_packages | 0 |
AI Readiness
How well is the repo equipped to be developed and maintained with AI coding agents? Carries a deliberately small weight (4%): agent tooling is a real maintenance signal, but a repository with none can still reach 100/100.
Agent context & guidance
How it's scored
| 0/45 | Agent instructions — no CLAUDE.md / AGENTS.md / editor rules |
| 0/15 | Machine-readable docs (llms.txt) |
| 27.2/40 | Legible commit history — 51 of 100 human commits state their intent (structured subject or explanatory body) |
Inputs used
| has_llms_txt | no |
| llms_txt_url | — |
| legible_history_share | 0.51 |
| agent_instruction_files | — |
| agent_instruction_max_bytes | — |
Verify loop (build / test / typecheck)
How it's scored
| 18/18 | One-command bootstrap — docs/Makefile, docs_nnx/Makefile |
| 22/22 | Automated tests |
| 11/11 | Lint / format config — pyproject.toml ([tool.ruff], [tool.yapf]) |
| 11/11 | Static type checking — flax/py.typed |
| 10/10 | Reproducible environment — lockfile |
| 0/10 | Demonstrated agent practice — no agent-authored commits among the last 100 |
| 0/8 | Automated maintenance — no automated dependency updates observed |
| 0/10 | OpenSSF Scorecard: Pinned-Dependencies — no data |
Inputs used
| has_nix | no |
| has_tests | yes |
| lockfiles | Cargo.lock, uv.lock |
| has_dockerfile | no |
| typed_language | no |
| bootstrap_files | docs/Makefile, docs_nnx/Makefile |
| has_devcontainer | no |
| has_linter_config | yes |
| typecheck_configs | flax/py.typed |
| agent_commit_share | 0 |
| toolchain_manifests | flaxlib_src/Cargo.toml |
| dependency_bot_commit_share | 0 |
Code legibility for models
How it's scored
| 27/45 | Type-checkable code — Jupyter Notebook with type-check config (flax/py.typed) |
| 53.3/55 | Manageable file sizes — 11/354 source files over 60KB |
Inputs used
| primary_language | Jupyter Notebook |
| largest_source_bytes | 233,046 |
| source_files_sampled | 354 |
| oversized_source_files | 11 |
Machine-readable interfaces
How it's scored
| 0/40 | API schema (OpenAPI/GraphQL/proto) — not applicable to this kind of software |
| 0/20 | MCP server — not applicable to this kind of software |
| 40/40 | Runnable examples — examples, notebooks |
Inputs used
| example_dirs | examples, notebooks |
| has_mcp_signal | no |
| api_schema_files | — |
| interfaces_expected_of | — |
Key facts
Data collection warnings
- Star history unavailable: GitHub GraphQL error: Resource not accessible by personal access token
- Could not fetch crates package 'flaxlib' from its registry
- OpenSSF Scorecard did not return a usable result (killed by SIGKILL — most likely the container's memory limit); skipping Scorecard checks
More detail
Star and fork history 0 ★ / 829 ⇿
When each star and fork was added, collected from GitHub and bucketed by day. Cumulative growth sits directly above the daily additions it is made of, so the two read against each other: steady organic accretion looks nothing like an abrupt, short-lived burst. Where that difference is measurable, it is reported as growth authenticity.
Each point covers 7 days.
Direct dependencies 11
| Registry | Package | Version constraint | Manifest |
|---|---|---|---|
| crates.io | pyo3 | 0.21.2 | flaxlib_src/Cargo.toml |
| PyPI | numpy | >=1.23.2 | pyproject.toml |
| PyPI | jax | >=0.11.1 | pyproject.toml |
| PyPI | msgpack | — | pyproject.toml |
| PyPI | optax | — | pyproject.toml |
| PyPI | orbax-checkpoint | — | pyproject.toml |
| PyPI | tensorstore | — | pyproject.toml |
| PyPI | rich | >=11.1 | pyproject.toml |
| PyPI | typing_extensions | >=4.2 | pyproject.toml |
| PyPI | PyYAML | >=5.4.1 | pyproject.toml |
| PyPI | treescope | >=0.1.7 | pyproject.toml |
All dependencies 112
Full resolved dependency set from the GitHub dependency graph: 17 direct and 95 indirect (transitive) packages. The transitive closure is complete when the repository commits a lockfile.
| Registry | Package | Version | Relation |
|---|---|---|---|
| crates.io | pyo3 | 0.21.2 | direct |
| PyPI | jax | — | direct |
| PyPI | jax | 0.3.4 | direct |
| PyPI | jax | 0.4.13 | direct |
| PyPI | numpy | — | direct |
| PyPI | numpy | 1.22.0 | direct |
| PyPI | numpy | 1.23.5 | direct |
| PyPI | numpy | 1.24.3 | direct |
| PyPI | optax | — | direct |
| PyPI | optax | 0.1.0 | direct |
| PyPI | optax | 0.1.3 | direct |
| PyPI | optax | 0.1.5 | direct |
| PyPI | orbax-checkpoint | — | direct |
| PyPI | pyyaml | — | direct |
| PyPI | rich | — | direct |
| PyPI | treescope | — | direct |
| PyPI | typing-extensions | — | direct |
| crates.io | autocfg | 1.3.0 | indirect |
| crates.io | bitflags | 2.6.0 | indirect |
| crates.io | cfg-if | 1.0.0 | indirect |
| crates.io | heck | 0.4.1 | indirect |
| crates.io | indoc | 2.0.5 | indirect |
| crates.io | libc | 0.2.158 | indirect |
| crates.io | lock_api | 0.4.12 | indirect |
| crates.io | memoffset | 0.9.1 | indirect |
| crates.io | once_cell | 1.20.0 | indirect |
| crates.io | parking_lot | 0.12.3 | indirect |
| crates.io | parking_lot_core | 0.9.10 | indirect |
| crates.io | portable-atomic | 1.7.0 | indirect |
| crates.io | proc-macro2 | 1.0.86 | indirect |
| crates.io | pyo3-build-config | 0.21.2 | indirect |
| crates.io | pyo3-ffi | 0.21.2 | indirect |
| crates.io | pyo3-macros | 0.21.2 | indirect |
| crates.io | pyo3-macros-backend | 0.21.2 | indirect |
| crates.io | quote | 1.0.37 | indirect |
| crates.io | redox_syscall | 0.5.7 | indirect |
| crates.io | scopeguard | 1.2.0 | indirect |
| crates.io | smallvec | 1.13.2 | indirect |
| crates.io | syn | 2.0.77 | indirect |
| crates.io | target-lexicon | 0.12.16 | indirect |
| crates.io | unicode-ident | 1.0.13 | indirect |
| crates.io | unindent | 0.2.3 | indirect |
| crates.io | windows-targets | 0.52.6 | indirect |
| crates.io | windows_aarch64_gnullvm | 0.52.6 | indirect |
| crates.io | windows_aarch64_msvc | 0.52.6 | indirect |
| crates.io | windows_i686_gnu | 0.52.6 | indirect |
| crates.io | windows_i686_gnullvm | 0.52.6 | indirect |
| crates.io | windows_i686_msvc | 0.52.6 | indirect |
| crates.io | windows_x86_64_gnu | 0.52.6 | indirect |
| crates.io | windows_x86_64_gnullvm | 0.52.6 | indirect |
| crates.io | windows_x86_64_msvc | 0.52.6 | indirect |
| PyPI | absl-py | — | indirect |
| PyPI | absl-py | 1.0.0 | indirect |
| PyPI | absl-py | 1.4.0 | indirect |
| PyPI | atari-py | 0.2.5 | indirect |
| PyPI | cloudpickle | — | indirect |
| PyPI | clu | 0.0.12 | indirect |
| PyPI | clu | 0.0.6 | indirect |
| PyPI | clu | 0.0.9 | indirect |
| PyPI | datasets | — | indirect |
| PyPI | dm-haiku | — | indirect |
| PyPI | flax | — | indirect |
| PyPI | flax | 0.3.6 | indirect |
| PyPI | flax | 0.4.1 | indirect |
| PyPI | flax | 0.6.0 | indirect |
| PyPI | flax | 0.6.11 | indirect |
| PyPI | flax | 0.6.5 | indirect |
| PyPI | flax | 0.6.9 | indirect |
| PyPI | google-benchmark | — | indirect |
| PyPI | google-cloud-storage | — | indirect |
| PyPI | grain | — | indirect |
| PyPI | grain | 0.2.16 | indirect |
| PyPI | gym | 0.18.3 | indirect |
| PyPI | ipywidgets | — | indirect |
| PyPI | jaxlib | 0.3.2+cuda11.cudnn82 | indirect |
| PyPI | jaxlib | 0.4.13+cuda11.cudnn82 | indirect |
| PyPI | jaxtyping | — | indirect |
| PyPI | jraph | — | indirect |
| PyPI | jraph | 0.0.2.dev0 | indirect |
| PyPI | jupytext | 1.13.8 | indirect |
| PyPI | kagglehub | — | indirect |
| PyPI | keras | — | indirect |
| PyPI | matplotlib | — | indirect |
| PyPI | ml-collections | — | indirect |
| PyPI | ml-collections | 0.1.0 | indirect |
| PyPI | ml-collections | 0.1.1 | indirect |
| PyPI | mlcroissant | — | indirect |
| PyPI | nanobind | — | indirect |
| PyPI | opencv-python | 4.5.4.60 | indirect |
| PyPI | orbax-export | — | indirect |
| PyPI | pillow | 10.2.0 | indirect |
| PyPI | pre-commit | — | indirect |
| PyPI | pygments | — | indirect |
| PyPI | scikit-build-core | — | indirect |
| PyPI | sentencepiece | — | indirect |
| PyPI | sentencepiece | 0.1.96 | indirect |
| PyPI | sentencepiece | 0.1.99 | indirect |
| PyPI | sentencepiece | 0.2.0 | indirect |
| PyPI | six | 1.15.0 | indirect |
| PyPI | sklearn | 0.0 | indirect |
| PyPI | sphinx | 6.2.1 | indirect |
| PyPI | tensorflow | 2.11.1 | indirect |
| PyPI | tensorflow | 2.12.0 | indirect |
| PyPI | tensorflow | 2.13.0 | indirect |
| PyPI | tensorflow-cpu | — | indirect |
| PyPI | tensorflow-datasets | — | indirect |
| PyPI | tensorflow-datasets | 4.4.0 | indirect |
| PyPI | tensorflow-datasets | 4.9.2 | indirect |
| PyPI | tensorflow-text | 2.13.0 | indirect |
| PyPI | tensorflow-text | 2.7.0 | indirect |
| PyPI | tensorflow-text | 2.8.1 | indirect |
| PyPI | tqdm | 4.67.1 | indirect |
Dependency advisories 0
Installing pypi:flax@0.12.9 pulls in 28 packages, direct and transitive: 0 carry known advisories, of which 0 are direct dependencies.
No known advisories affect the assessed dependencies.
An advisory means the version recorded in the dependency graph falls inside an advisory’s affected range. Reachability is not analysed, and the graph includes development and test pins — a finding may concern tooling rather than shipped software.
Raw JSON report machine-readable
Feedback
Spotted something off in this report, or have thoughts to share? Wrong measurements, missed tooling, ideas, questions — anything is welcome. Every message is read and gets a response.
Scores are signals, not warranties. They reflect publicly visible practices on GitHub — not a code audit, and not a security guarantee.
Missing data is excluded and weights renormalized, never scored as zero. Methodology is versioned and open: metrics v2.10.0, schema v0.34.0 — full methodology · metrics wiki.
How one result sits in the wider record: aggregate statistics — PyPI.