google/flax

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

[view on github]last commit: Oct 12, 2026

stars

7,348

7d

+6

30d

+23

90d

-

## star history

## found in

Awesome Open Source AI/Deep Learning Frameworks