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

flax

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

7,303stars

Forks

830

Open issues

510

Watchers

7,303

Size

44.7 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: Aug 25, 2026

Last push: Aug 24, 2026