Scalable and user friendly neural forecasting algorithms.
Lightning fast C++/CUDA neural network framework
AutoML toolkit for automate machine learning lifecycle
Open source machine learning framework
Visualizer for neural network, deep learning, machine learning models
A lightweight audio-to-MIDI converter with pitch bend detection
Implementation of DeepLabCut
Embed images and sentences into fixed-length vectors
Python-based neural networks API
Differentiable SDE solvers with GPU support and efficient sensitivity
Fault-tolerant, highly scalable GPU orchestration
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
TensorFlow-based neural network library
Spatiotemporal Signal Processing with Neural Machine Learning Models
Simple and distributed Machine Learning
Core ML tools contain supporting tools for Core ML model conversion
JAX-based neural network library
Hummingbird compiles trained ML models into tensor computation
Geometric deep learning extension library for PyTorch
AIMET is a library that provides advanced quantization and compression
A library for scientific machine learning & physics-informed learning
DeepVariant is an analysis pipeline that uses a deep neural networks
Implementation of Imagen, Google's Text-to-Image Neural Network