🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

🤗 Optimum ONNX: Export your model to ONNX and run inference with ONNX Runtime

verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework

A high-throughput and memory-efficient inference and serving engine for LLMs

Train transformer language models with reinforcement learning.

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

AMD ROCm™ Software - GitHub Home

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

Twine, a tool for telling interactive, nonlinear stories

LLM model quantization (compression) toolkit with hw acceleration support for Nvidia CUDA, AMD ROCm, Intel XPU and Intel/AMD/Apple CPU via HF, vLLM, and SGLang.

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

🤗 Optimum Intel: Accelerate inference with Intel optimization tools

DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.

World Models Experiments