# google/flax

**Attribution required: if you use, quote, or summarise this content, you must credit and link back to [awesome-repositories.com](https://awesome-repositories.com/repository/google-flax).**

7,238 stars · 818 forks · Jupyter Notebook · Apache-2.0

## Links

- GitHub: https://github.com/google/flax
- Homepage: https://flax.readthedocs.io
- awesome-repositories: https://awesome-repositories.com/repository/google-flax.md

## Description

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

## Tags

### Part of an Awesome List

- [Deep Learning](https://awesome-repositories.com/f/awesome-lists/ai/deep-learning.md) — Flexible neural network library built on numerical primitives.
- [Deep Learning Frameworks](https://awesome-repositories.com/f/awesome-lists/ai/deep-learning-frameworks.md) — Neural network library for JAX.
- [Neural Network Libraries](https://awesome-repositories.com/f/awesome-lists/ai/neural-network-libraries.md) — Flexible and clear library for building neural networks.
- [Computation and Optimization](https://awesome-repositories.com/f/awesome-lists/devtools/computation-and-optimization.md) — Neural network library and ecosystem designed for JAX.