google/flax

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

Jupyter Notebook7,204 stars810 forksApache-2.0

jax

Rankings

Daily

#6336

Weekly

#10673

Monthly

#11423

Total stars

Daily new stars

Get badge

This repository is tracked by Trending Repos. The badge upgrades automatically if it ever cracks the top 100.

Markdown

![Trending Repos](https://trending-repos.com/badge/google/flax.svg)

HTML

<img src="https://trending-repos.com/badge/google/flax.svg" alt="Trending Repos" />

Raw URL

https://trending-repos.com/badge/google/flax.svg