AutoML toolkit for automate machine learning lifecycle
Lightning fast C++/CUDA neural network framework
Scalable and user friendly neural forecasting algorithms.
Visualizer for neural network, deep learning, machine learning models
Open source machine learning framework
C++ library for high performance inference on NVIDIA GPUs
A lightweight audio-to-MIDI converter with pitch bend detection
Python-based neural networks API
A scikit-learn compatible neural network library that wraps PyTorch
Simulation of spiking neural networks (SNNs) using PyTorch
Hummingbird compiles trained ML models into tensor computation
Implementation of DeepLabCut
OpenMMLab Model Deployment Framework
TensorFlow-based neural network library
Graph Neural Network Library for PyTorch
AI discovers 520000 stable inorganic crystal structures for research
Deep learning library
A Python package for extending the official PyTorch
JAX-based neural network library
Open standard for machine learning interoperability
Implementation of Imagen, Google's Text-to-Image Neural Network
Message Passing Neural Networks for Molecule Property Prediction
A library for scientific machine learning & physics-informed learning
AIMET is a library that provides advanced quantization and compression
Implementation of 'lightweight' GAN, proposed in ICLR 2021