Synthetic data generators for tabular and time-series data
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
A comprehensive set of fairness metrics for datasets
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Embed images and sentences into fixed-length vectors
Geometric deep learning extension library for PyTorch
Technical principles related to large models
A natural language interface for computers
The largest collection of PyTorch image encoders / backbones
Opensource browser using agents
Benchmarking Multimodal Agents for Open-Ended Tasks
[NeurIPS 2023 Spotlight] LightZero
Advanced evolutionary computation library built on top of PyTorch
Implementation of RLHF (Reinforcement Learning with Human Feedback)
An Easy-to-use, Scalable and High-performance RLHF Framework
LLM abstractions that aren't obstructions
Standalone, small, language-neutral
TFX is an end-to-end platform for deploying production ML pipelines
SkyPilot: Run AI and batch jobs on any infra
Interpretable prompting and models for NLP
Library of self-supervised methods for visual representation
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
An open source object detection toolbox based on PyTorch