JAX-based neural network library
A cross-platform Python library for differentiable programming
Graph Neural Network Library for PyTorch
Flax is a neural network library for JAX
Implementation of Imagen, Google's Text-to-Image Neural Network
TensorFlow-based neural network library
Training and serving large-scale neural networks
Neural network library for developers