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