Overview | Quick install | What does Flax look like? | Documentation This README is a very short intro. To learn everything you need to know about Flax, refer to our full documentation. Flax was originally started by engineers and researchers within the Brain Team in Google Research (in close collaboration with the JAX team), and is now developed jointly with the open source community. Flax is bei