Implementation of Imagen, Google's Text-to-Image Neural Network
A cross-platform Python library for differentiable programming
JAX-based neural network library
AIMET is a library that provides advanced quantization and compression
A scikit-learn compatible neural network library that wraps PyTorch
Python Optimal Transport
Core ML tools contain supporting tools for Core ML model conversion
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
A library for scientific machine learning & physics-informed learning
A lightweight audio-to-MIDI converter with pitch bend detection
Hummingbird compiles trained ML models into tensor computation
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Decentralized deep learning in PyTorch. Built to train models
Simulation of spiking neural networks (SNNs) using PyTorch
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Time series forecasting with PyTorch
DeepVariant is an analysis pipeline that uses a deep neural networks
TensorFlow-based neural network library
MiniSom is a minimalistic implementation of the Self Organizing Maps
JAX-based neural network library
A curated list of data mining papers about fraud detection
AI discovers 520000 stable inorganic crystal structures for research
A Python package for extending the official PyTorch
A toolkit to optimize ML models for deployment for Keras & TensorFlow