PyTorch extensions for fast R&D prototyping and Kaggle farming
An easy-to-use & supercharged open-source experiment tracker
A simple but complete full-attention transformer
An Easy-to-use, Scalable and High-performance RLHF Framework
Reference implementations of MLPerf™ training benchmarks
JAX-based neural network library
AutoGluon: AutoML for Image, Text, and Tabular Data
A PyTorch-based Speech Toolkit
The most intuitive, flexible, way for researchers to build models
Open source framework for deep learning satellite and aerial imagery
Fast image augmentation library and an easy-to-use wrapper
A library for accelerating Transformer models on NVIDIA GPUs
Superfast AI decision making and processing of multi-modal data
A system for quickly generating training data with weak supervision
Build MLOps Pipelines in Minutes
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
A unified, comprehensive and efficient recommendation library
Statistical library designed to fill the void in Python's time series
An MLOps framework to package, deploy, monitor and manage models
Training data (data labeling, annotation, workflow) for all data types
Train machine learning models within Docker containers
TensorFlow-based neural network library
This repos contains notebooks for the Advanced Solutions Lab
Images to inference with no labeling
Your open-source LLM evaluation toolkit