Implementation of Imagen, Google's Text-to-Image Neural Network
Graph Neural Network Library for PyTorch
AIMET is a library that provides advanced quantization and compression
Training PyTorch models with differential privacy
A cross-platform Python library for differentiable programming
A JAX research toolkit to build, edit, & visualize neural networks
Flax is a neural network library for JAX
Facebook AI Research Sequence-to-Sequence Toolkit written in Python
A model library for exploring state-of-the-art deep learning
Tools to help users inter-operate among deep learning frameworks
Basic Utilities for PyTorch Natural Language Processing (NLP)
Pretty Tensor: Fluent Networks in TensorFlow