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

flax

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

7,158stars

Forks

796

Open issues

474

Watchers

7,158

Size

32.9 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: Apr 14, 2026

Last push: Apr 13, 2026