⭐ Star AlbumentationsX on GitHub — 448+ stars and counting!

flax

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

7,212stars

Forks

812

Open issues

476

Watchers

7,212

Size

37.9 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: May 28, 2026

Last push: May 22, 2026