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