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

flax

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

7,340stars

Forks

847

Open issues

534

Watchers

7,340

Size

44.7 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: Oct 12, 2026

Last push: Oct 8, 2026