google

google/flax

steady

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

Jupyter Notebook View on GitHub

Stars

7,271

Forks

825

Open issues

228

24h

0

0.0%

7d

+11

+0.2%

Refresh

2h

Star history (7 days)

Last checked

23m ago

Last pushed

07 Jul 2026

Next check

just now