Scalable and user friendly neural forecasting algorithms.
Instant neural graphics primitives: lightning fast NeRF and more
Lightning fast C++/CUDA neural network framework
Pre-built implicit layer architectures with O(1) backprop, GPUs
Message Passing Neural Networks for Molecule Property Prediction
Simulation of spiking neural networks (SNNs) using PyTorch
Visualizer for neural network, deep learning, machine learning models
oneAPI Deep Neural Network Library (oneDNN)
Physics-Informed Neural Networks (PINN) Solvers
Training neural networks on Apple Neural Engine via APIs
Open source machine learning framework
Nx-powered Neural Networks
AIMET is a library that provides advanced quantization and compression
Unsupervised text tokenizer for Neural Network-based text generation
BitNet: Scaling 1-bit Transformers for Large Language Models
A scikit-learn compatible neural network library that wraps PyTorch
Sandbox for training deep learning networks
JAX-based neural network library
Implementation of Imagen, Google's Text-to-Image Neural Network
A general-purpose probabilistic programming system
Time series forecasting with PyTorch
DNN && GAN && NLP && BIG DATA
A generic, simple and fast implementation of Deepmind's AlphaZero
DeepVariant is an analysis pipeline that uses a deep neural networks
Graph Neural Network Library for PyTorch