Repositories list

• Causal-Forcing

  Public

  thu-ml/Causal-Forcing’s past year of commit activity
  Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation"

  diffusiondistillationvideo-generation+ 14

  diffusiondistillationvideo-generationworld-modelsdiffusion-modelsconsistency-modelstext-to-videoautoregressive-modelsgenerative-aiworld-model+ 7

  Python

  •

  Apache License 2.0

  •3535 forks•580580 stars•1717 issues•00 pull requests•Updated Apr 23, 2026Apr 23, 2026

• TurboDiffusion

  Public

  thu-ml/TurboDiffusion’s past year of commit activity
  TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

  rcmdistillationvideo-generation+ 7

  rcmdistillationvideo-generationmlsystemdiffusion-modelsinference-accelerationai-infraconsistency-modelsageattentionsparse-linear-attention

  Python

  •

  Apache License 2.0

  •251251 forks•3.5k3.5k stars•6767 issues•77 pull requests•Updated Apr 15, 2026Apr 15, 2026

• tianshou

  Public

  thu-ml/tianshou’s past year of commit activity
  An elegant PyTorch deep reinforcement learning library.

  pytorchdqnpolicy-gradient+ 16

  pytorchdqnpolicy-gradientrlcqlatariddpgimitation-learningsacdrl+ 9

  Python

  •

  MIT License

  •1.3k1.3k forks•11k11k stars•133133 issues•44 pull requests•Updated Apr 3, 2026Apr 3, 2026

• Auto-MIA

  Public

  thu-ml/Auto-MIA’s past year of commit activity
  Python

  •00 forks•33 stars•00 issues•00 pull requests•Updated Mar 24, 2026Mar 24, 2026

• RGBT-Clothing

  Public

  thu-ml/RGBT-Clothing’s past year of commit activity
  Python

  •00 forks•33 stars•00 issues•00 pull requests•Updated Mar 24, 2026Mar 24, 2026

• embodied-data-toolkit

  Public
  A toolkit for processing raw embodied data into standardized formats and converting between embodied dataset schemas.

  data-processingembodied-aiformat-converter-tool

  data-processingembodied-aiformat-converter-tool

  Python

  •22 forks•1717 stars•00 issues•00 pull requests•Updated Mar 16, 2026Mar 16, 2026

• cond-image-leakage

  Public

  thu-ml/cond-image-leakage’s past year of commit activity
  Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)

  Apache License 2.0

  •2222 forks•257257 stars•88 issues•00 pull requests•Updated Mar 13, 2026Mar 13, 2026

• DiT-Extrapolation

  Public

  thu-ml/DiT-Extrapolation’s past year of commit activity
  Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) , UltraViCo (ICLR 2026) and UltraImage

  generative-modelropediffusion+ 9

  generative-modelropediffusionextrapolationditvideo-generationposition-embeddingdiffusion-modelslong-video-generationdiffusion-transformer+ 2

  Python

  •

  Apache License 2.0

  •7575 forks•806806 stars•2424 issues•00 pull requests•Updated Mar 8, 2026Mar 8, 2026

• SpargeAttn

  Public

  thu-ml/SpargeAttn’s past year of commit activity
  [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

  attentionvitquantization+ 8

  attentionvitquantizationvideo-generationmlsysinference-accelerationai-infravision-transformersparse-attentionllm+ 1

  Cuda

  •

  Apache License 2.0

  •9191 forks•985985 stars•5858 issues•22 pull requests•Updated Feb 25, 2026Feb 25, 2026

• SLA

  Public

  thu-ml/SLA’s past year of commit activity
  SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention

  transformervideo-generationmlsys+ 7

  transformervideo-generationmlsysinference-accelerationai-infralinear-attentionsparse-attentiondiffusion-transformertrain-accelerationsparse-linear-attention

  Python

  •

  Apache License 2.0

  •1818 forks•299299 stars•99 issues•00 pull requests•Updated Feb 24, 2026Feb 24, 2026

• CausalForcing.github.io

  Public
  JavaScript

  •00 forks•00 stars•00 issues•00 pull requests•Updated Feb 7, 2026Feb 7, 2026

• RDT2

  Public

  thu-ml/RDT2’s past year of commit activity
  Official code of RDT 2

  Python

  •

  Apache License 2.0

  •4747 forks•758758 stars•1616 issues•22 pull requests•Updated Feb 7, 2026Feb 7, 2026

• ultraimage.github.io

  Public

  thu-ml/ultraimage.github.io’s past year of commit activity
  JavaScript

  •00 forks•00 stars•00 issues•00 pull requests•Updated Feb 6, 2026Feb 6, 2026

• i-DODE

  Public
  Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)

  Python

  •

  Apache License 2.0

  •11 fork•1919 stars•11 issue•00 pull requests•Updated Jan 27, 2026Jan 27, 2026

• RoboticsDiffusionTransformer

  Public

  thu-ml/RoboticsDiffusionTransformer’s past year of commit activity
  RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation

  Python

  •

  MIT License

  •156156 forks•1.7k1.7k stars•3838 issues•33 pull requests•Updated Jan 21, 2026Jan 21, 2026

• SageAttention

  Public

  thu-ml/SageAttention’s past year of commit activity
  [ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across la…

  cudatritonattention+ 9

  cudatritonattentionvitquantizationvideo-generationmlsysinference-accelerationefficient-attentionllm+ 2

  Cuda

  •

  Apache License 2.0

  •399399 forks•3.3k3.3k stars•161161 issues•1919 pull requests•Updated Jan 17, 2026Jan 17, 2026

• MLA-Trust

  Public
  A toolbox for benchmarking Multimodal LLM Agents trustworthiness across truthfulness, controllability, safety and privacy dimensions through 34 interactive task…

  agentbenchmarkprivacy+ 6

  agentbenchmarkprivacytoolboxsafetymulti-modalcontrollabilitytrustworthy-aitruthfulness

  Python

  •

  MIT License

  •44 forks•6363 stars•33 issues•00 pull requests•Updated Jan 9, 2026Jan 9, 2026

• Motus

  Public
  Official code of Motus: A Unified Latent Action World Model

  roboticsdiffusion-modelvideo-generation+ 4

  roboticsdiffusion-modelvideo-generationrobotic-manipulationworld-modelunidiffuservision-language-action-model

  Python

  •

  Apache License 2.0

  •4747 forks•990990 stars•2323 issues•11 pull request•Updated Jan 5, 2026Jan 5, 2026

• vidar-robotwin

  Public

  thu-ml/vidar-robotwin’s past year of commit activity
  robotwin evaluation code for vidar.

  Python

  •

  MIT License

  •00 forks•55 stars•11 issue•00 pull requests•Updated Dec 22, 2025Dec 22, 2025

• vidar

  Public
  Official repo for vidar and vidarc: video foundation model for robotics.

  Python

  •11 fork•4040 stars•11 issue•00 pull requests•Updated Dec 22, 2025Dec 22, 2025

• UltraViCo.github.io

  Public
  Project page for "UltraViCo"

  JavaScript

  •00 forks•00 stars•00 issues•00 pull requests•Updated Dec 3, 2025Dec 3, 2025

• UniCardio

  Public
  Versatile Cardiovascular Signal Generation with a Unified Diffusion Transformer

  Python

  •

  MIT License

  •44 forks•5656 stars•22 issues•00 pull requests•Updated Oct 9, 2025Oct 9, 2025

• FrameBridge

  Public
  FrameBridge: Improving Image-to-Video Generation with Bridge Models

  Python

  •00 forks•33 stars•00 issues•00 pull requests•Updated Oct 2, 2025Oct 2, 2025

• DiffusionBridge

  Public
  Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)

  Python

  •1010 forks•112112 stars•66 issues•00 pull requests•Updated Sep 9, 2025Sep 9, 2025

• SparseDM

  Public

  thu-ml/SparseDM’s past year of commit activity
  Python

  •11 fork•11 star•00 issues•00 pull requests•Updated Aug 24, 2025Aug 24, 2025

• EmbodiedActiveDefense

  Public
  Official implementation for "Reinforced Embodied Active Defense: Exploiting Adaptive Interaction for Robust Visual Perception in Adversarial 3D Environments" (T…

  Python

  •

  MIT License

  •00 forks•88 stars•00 issues•00 pull requests•Updated Jul 26, 2025Jul 26, 2025

• oddefense

  Public

  thu-ml/oddefense’s past year of commit activity
  Jupyter Notebook

  •

  MIT License

  •22 forks•66 stars•00 issues•00 pull requests•Updated Jul 16, 2025Jul 16, 2025

• Adaptive-Sparse-Trainer

  Public

  thu-ml/Adaptive-Sparse-Trainer’s past year of commit activity
  Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)

  Python

  •

  Apache License 2.0

  •22 forks•1919 stars•00 issues•00 pull requests•Updated Jul 1, 2025Jul 1, 2025

• MMTrustEval

  Public

  thu-ml/MMTrustEval’s past year of commit activity
  A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)

  benchmarkprivacytoolbox+ 9

  benchmarkprivacytoolboxsafetymulti-modalfairnessrobustnessclaudegpt-4trustworthy-ai+ 2

  Python

  •

  Creative Commons Attribution Share Alike 4.0 International

  •1212 forks•173173 stars•11 issue•00 pull requests•Updated Jun 27, 2025Jun 27, 2025

• TetraJet-MXFP4Training

  Public

  thu-ml/TetraJet-MXFP4Training’s past year of commit activity
  Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training

  Python

  •

  Apache License 2.0

  •44 forks•3939 stars•00 issues•00 pull requests•Updated Jun 20, 2025Jun 20, 2025