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

flax

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

7,264stars

Forks

824

Open issues

493

Watchers

7,264

Size

41.9 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: Jul 12, 2026

Last push: Jul 7, 2026