1google / flaxFlax is a neural network library for JAX that is designed for flexibility.jaxJupyter NotebookAPACHE-2.07.4K total849 forks+7 starsfrom 7,356 · score 2.08