JAX - A curated list of resources https://github.com/google/jax
Deep learning operations reinvented (for pytorch, tensorflow, chainer, gluon and others)
Flax is a neural network library for JAX that is designed for flexibility.
A collection of useful .gitignore templates
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Development site and blog for Quansight Labs
All about the fundamental blocks of TF and JAX!
Mathematical operations for JAX pytrees