Official implementation of AnimateDiff.

[CVPR 2025 Best Paper Award] VGGT: Visual Geometry Grounded Transformer

Wan: Open and Advanced Large-Scale Video Generative Models

HunyuanVideo: A Systematic Framework For Large Video Generation Model

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]

Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 500+ LLMs (Qwen3, Qwen3-MoE, Llama4, GLM4.5, InternLM3, DeepSeek-R1, ...) and 200+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, Llava, GLM4v, Ph…

PyTorch package for the discrete VAE used for DALL·E.

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

Hackable and optimized Transformers building blocks, supporting a composable construction.

Refine high-quality datasets and visual AI models

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

🚀 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

[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, 17+ clouds, or on-prem).

Kronos: A Foundation Model for the Language of Financial Markets

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Python client for Baidu Yun (Personal Cloud Storage) 百度云/百度网盘Python客户端

An Easy-to-use, Scalable and High-performance RLHF Framework based on Ray (PPO & GRPO & REINFORCE++ & vLLM & Ray & Dynamic Sampling & Async Agentic RL)

StarCraft II Learning Environment

A PyTorch implementation of EfficientNet

Pytorch implementation of convolutional neural network visualization techniques

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

tensorboard for pytorch (and chainer, mxnet, numpy, ...)

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM

A faster pytorch implementation of faster r-cnn