google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Stars: 7,095Language: Jupyter Notebook
Give AlbumentationsX a star on GitHub — it powers this leaderboard
Star on GitHubFlax is a neural network library for JAX that is designed for flexibility.