A library for scientific machine learning & physics-informed learning
The Python Code Tutorials
Hummingbird compiles trained ML models into tensor computation
Core ML tools contain supporting tools for Core ML model conversion
Implementation of Imagen, Google's Text-to-Image Neural Network
JAX-based neural network library
A scikit-learn compatible neural network library that wraps PyTorch
Simulation of spiking neural networks (SNNs) using PyTorch
Easy-to-use,Modular and Extendible package of deep-learning models
Graph Neural Network Library for PyTorch
DeepVariant is an analysis pipeline that uses a deep neural networks
Implementation of 'lightweight' GAN, proposed in ICLR 2021
SpikingJelly is an open-source deep learning framework
Training PyTorch models with differential privacy
Package of deep-learning based CTR models
Time series forecasting with PyTorch
A cross-platform Python library for differentiable programming
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Decentralized deep learning in PyTorch. Built to train models
AIMET is a library that provides advanced quantization and compression
Python Optimal Transport
JAX-based neural network library
Foundation Model for Tabular Data
Deep learning driven jazz generation using Keras & Theano
High-Performance Face Recognition Library on PaddlePaddle & PyTorch