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
Unsupervised text tokenizer for Neural Network-based text generation
A lightweight audio-to-MIDI converter with pitch bend detection
Python-based neural networks API
A Python package for extending the official PyTorch
Message Passing Neural Networks for Molecule Property Prediction
Simulation of spiking neural networks (SNNs) using PyTorch
A scikit-learn compatible neural network library that wraps PyTorch
A library for scientific machine learning & physics-informed learning
Hummingbird compiles trained ML models into tensor computation
OpenMMLab Model Deployment Framework
Simple and distributed Machine Learning
TensorFlow-based neural network library
Graph Neural Network Library for PyTorch
Implementation of DeepLabCut
JAX-based neural network library
Deep learning library
Implementation of Imagen, Google's Text-to-Image Neural Network
ktrain is a Python library that makes deep learning AI more accessible
AIMET is a library that provides advanced quantization and compression
Implementation of 'lightweight' GAN, proposed in ICLR 2021