A library for scientific machine learning & physics-informed learning
Flax is a neural network library for JAX
A JAX research toolkit to build, edit, & visualize neural networks
Implementation of Imagen, Google's Text-to-Image Neural Network
Convolutional neural network model for video classification
Pretty Tensor: Fluent Networks in TensorFlow