Open deep learning compiler stack for cpu, gpu, etc.
PyTorch version of Stable Baselines
An Easy-to-use, Scalable and High-performance RLHF Framework
Reference implementations of MLPerf™ training benchmarks
A simple but complete full-attention transformer
A game theoretic approach to explain the output of ml models
Implementation of Imagen, Google's Text-to-Image Neural Network
JAX-based neural network library
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
PyTorch extensions for fast R&D prototyping and Kaggle farming
Statistical library designed to fill the void in Python's time series
A library for accelerating Transformer models on NVIDIA GPUs
AutoGluon: AutoML for Image, Text, and Tabular Data
A PyTorch-based Speech Toolkit
The most intuitive, flexible, way for researchers to build models
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
Open source framework for deep learning satellite and aerial imagery
Fast image augmentation library and an easy-to-use wrapper
Build MLOps Pipelines in Minutes
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
A unified, comprehensive and efficient recommendation library
This repos contains notebooks for the Advanced Solutions Lab
Images to inference with no labeling
TensorFlow-based neural network library