Star AlbumentationsX on GitHub — it powers this leaderboard

← Back to leaderboard

google

flax

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

7,091stars

Homepage View on GitHub

Forks

793

Open issues

477

Watchers

7,091

Size

32.3 MB

Jupyter NotebookApache License 2.0

jax

Created: Jan 10, 2020

Updated: Feb 28, 2026

Last push: Feb 27, 2026