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