Hyperparameter Optimization for TensorFlow, Keras and PyTorch
Open deep learning compiler stack for cpu, gpu, etc.
End-to-End Library for Continual Learning based on PyTorch
PyTorch version of Stable Baselines
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Determined, deep learning training platform
AI Toolkit for Healthcare Imaging
Toolkit for conversational AI
Learning agent trained in a diffusion world model
Scalable and user friendly neural forecasting algorithms.
AutoGluon: AutoML for Image, Text, and Tabular Data
A game theoretic approach to explain the output of ml models
An unsupervised and free tool for image and video dataset analysis
A library for deep learning end-to-end dialog systems and chatbots
Deep learning library
A library for accelerating Transformer models on NVIDIA GPUs
A Python library for audio
The Unified Machine Learning Framework
Python toolbox to create adversarial examples
Python library for defining and optimizing mathematical expressions
Data loaders and abstractions for text and NLP
Geometric deep learning extension library for PyTorch
Jittor is a high-performance deep learning framework
Smart Thermodynamic Modeling with Graph Neural Networks