Lightning fast C++/CUDA neural network framework
Instant neural graphics primitives: lightning fast NeRF and more
Visualizer for neural network, deep learning, machine learning models
JAX-based neural network library
BitNet: Scaling 1-bit Transformers for Large Language Models
oneAPI Deep Neural Network Library (oneDNN)
Simulation of spiking neural networks (SNNs) using PyTorch
Sandbox for training deep learning networks
A scikit-learn compatible neural network library that wraps PyTorch
Hummingbird compiles trained ML models into tensor computation
Graph Neural Network Library for PyTorch
A library for scientific machine learning & physics-informed learning
Nx-powered Neural Networks
Training neural networks on Apple Neural Engine via APIs
Implementation of Imagen, Google's Text-to-Image Neural Network
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
DNN && GAN && NLP && BIG DATA
Unsupervised text tokenizer for Neural Network-based text generation
A generic, simple and fast implementation of Deepmind's AlphaZero
DeepVariant is an analysis pipeline that uses a deep neural networks
Time series forecasting with PyTorch
A general-purpose probabilistic programming system
C++ library for high performance inference on NVIDIA GPUs
Package of deep-learning based CTR models
Deep learning driven jazz generation using Keras & Theano