Timestep Embedding Tells: It's Time to Cache for Video Diffusion Model

A flexible and efficient training framework for large-scale alignment tasks

An AI Hedge Fund Team

A generative world for general-purpose robotics & embodied AI learning.

[CVPR 2024 Oral] InternVL Family: A Pioneering Open-Source Alternative to GPT-4o. 接近GPT-4o表现的开源多模态对话模型

Code for BLT research paper

Code and models for NExT-GPT: Any-to-Any Multimodal Large Language Model

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation

Efficient LLM Inference over Long Sequences

O1 Replication Journey: A Strategic Progress Report – Part I

Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning

A debugging and profiling tool that can trace and visualize python code execution

Official repository of "SAMURAI: Adapting Segment Anything Model for Zero-Shot Visual Tracking with Motion-Aware Memory"

Official PyTorch implementation of SegFormer

Implementation of SmoothCache, a project aimed at speeding-up Diffusion Transformer (DiT) based GenAI models with error-guided caching.

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)

A lightweight memory allocator for hardware-accelerated machine learning

A PyTorch native library for large model training

[CVPR 2023] DepGraph: Towards Any Structural Pruning

PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538

AlphaFold 3 inference pipeline.

SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models

Run your own AI cluster at home with everyday devices 📱💻 🖥️⌚

The official GitHub page for the survey paper "A Survey on Mixture of Experts".

Puzzles for learning Triton

Stateful LLM Serving