A cross-platform Python library for differentiable programming
JAX-based neural network library
Industrial-strength Natural Language Processing (NLP)
Flax is a neural network library for JAX
A library for easy and efficient manipulation of tensor networks