Lightning fast C++/CUDA neural network framework
Visualizer for neural network, deep learning, machine learning models
Instant neural graphics primitives: lightning fast NeRF and more
JAX-based neural network library
oneAPI Deep Neural Network Library (oneDNN)
Simulation of spiking neural networks (SNNs) using PyTorch
A scikit-learn compatible neural network library that wraps PyTorch
Graph Neural Network Library for PyTorch
DNN && GAN && NLP && BIG DATA
Hummingbird compiles trained ML models into tensor computation
A library for scientific machine learning & physics-informed learning
Training neural networks on Apple Neural Engine via APIs
Nx-powered Neural Networks
Implementation of Imagen, Google's Text-to-Image Neural Network
Unsupervised text tokenizer for Neural Network-based text generation
A generic, simple and fast implementation of Deepmind's AlphaZero
A general-purpose probabilistic programming system
Deep learning driven jazz generation using Keras & Theano
PyTorch Getting Started Tutorial, read online
Training PyTorch models with differential privacy
Easy-to-use,Modular and Extendible package of deep-learning models
Package of deep-learning based CTR models
AIMET is a library that provides advanced quantization and compression
Time series forecasting with PyTorch
SpikingJelly is an open-source deep learning framework