Lightning fast C++/CUDA neural network framework
Flax is a neural network library for JAX
AIMET is a library that provides advanced quantization and compression
Stanford NLP Python library for many human languages
A library for scientific machine learning & physics-informed learning
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Graph Neural Network Library for PyTorch
Facebook AI Research Sequence-to-Sequence Toolkit written in Python
We provide a PyTorch implementation of the paper Voice Separation
Auto-diff neural network library for high-dimensional sparse tensors
Fast and flexible AutoML with learning guarantees
Non-local Neural Networks for Video Classification
PyTorch tutorials and fun projects including neural talk
Pretty Tensor: Fluent Networks in TensorFlow
This is an implementation of a machine learning library in C++17
Semantic image segmentation method described in the ICCV 2015 paper