Visualizer for neural network, deep learning, machine learning models
A generic, simple and fast implementation of Deepmind's AlphaZero
Implementation of Imagen, Google's Text-to-Image Neural Network
C++ library for high performance inference on NVIDIA GPUs
Unsupervised text tokenizer for Neural Network-based text generation
A cross-platform Python library for differentiable programming
JAX-based neural network library
AIMET is a library that provides advanced quantization and compression
A scikit-learn compatible neural network library that wraps PyTorch
Python Optimal Transport
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Core ML tools contain supporting tools for Core ML model conversion
Lightning fast C++/CUDA neural network framework
Nx-powered Neural Networks
A library for scientific machine learning & physics-informed learning
oneAPI Deep Neural Network Library (oneDNN)
Hummingbird compiles trained ML models into tensor computation
A lightweight audio-to-MIDI converter with pitch bend detection
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Decentralized deep learning in PyTorch. Built to train models
Simulation of spiking neural networks (SNNs) using PyTorch
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
TensorFlow-based neural network library