Sign in
← Back to search
github Actively maintained

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

2 awesome lists

Quick read

Stars
7,285
Forks
826
Open issues
497
Commits
5,902

Activity and growth

Latest capture 2026-08-03 03:03

Stars · last 7 days
No history
Commits · last 7 days
No history
Stars since tracking
+80
Stored snapshots
8

Classification

Topics

#jax

Technology stack

pytest

Metadata

Language
Jupyter Notebook
License
Apache-2.0
Default branch
main
Created
2020-01-10
First commit
2020-01-07
Last pushed
2026-08-01
GitHub updated
2026-08-02
Last synced
2026-08-03 03:03
Stack scanned
2026-08-03 03:03
Archived
No
Sign in to like google/flax
README Open website View on GitHub

AI development signals

0 paths

Agent instructions and tool configuration found in this repository.

No config files detected.

Growth history

Tracked growth

8 observed captures since 2026-05-22. Observed captures are shown by default.

Stars from first capture +80

Chart data

Observed captures only

Time horizon

All tracked data

Custom date range

Stars history

Observed snapshots

Commits history

Observed snapshots

Similar repositories

Nearest indexed repositories by embedding similarity.

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

36,252 stars
Python 3 awesome lists

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

2,936 stars
Python 1 awesome list

microsoft/FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

4,386 stars
Jupyter Notebook 1 awesome list

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

2,324 stars
Python 1 awesome list