Simulation of spiking neural networks (SNNs) using PyTorch
A scikit-learn compatible neural network library that wraps PyTorch
Lightning fast C++/CUDA neural network framework
Scalable and user friendly neural forecasting algorithms.
JAX-based neural network library
Hummingbird compiles trained ML models into tensor computation
Visualizer for neural network, deep learning, machine learning models
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Implementation of Imagen, Google's Text-to-Image Neural Network
Sandbox for training deep learning networks
AIMET is a library that provides advanced quantization and compression
BitNet: Scaling 1-bit Transformers for Large Language Models
Instant neural graphics primitives: lightning fast NeRF and more
Message Passing Neural Networks for Molecule Property Prediction
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
A library for scientific machine learning & physics-informed learning
A cross-platform Python library for differentiable programming
A generic, simple and fast implementation of Deepmind's AlphaZero
DeepVariant is an analysis pipeline that uses a deep neural networks
oneAPI Deep Neural Network Library (oneDNN)
Core ML tools contain supporting tools for Core ML model conversion
Deep learning driven jazz generation using Keras & Theano
Training PyTorch models with differential privacy
Package of deep-learning based CTR models