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

  •3333 forks•572572 stars•1717 issues•00 pull requests•Updated Apr 18, 2026Apr 18, 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

  •250250 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

  thu-ml/embodied-data-toolkit’s past year of commit activity
  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•804804 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•977977 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•298298 stars•99 issues•00 pull requests•Updated Feb 24, 2026Feb 24, 2026

• CausalForcing.github.io

  Public

  thu-ml/CausalForcing.github.io’s past year of commit activity
  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•757757 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

  thu-ml/i-DODE’s past year of commit activity
  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•159159 issues•1919 pull requests•Updated Jan 17, 2026Jan 17, 2026

• MLA-Trust

  Public

  thu-ml/MLA-Trust’s past year of commit activity
  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

  thu-ml/Motus’s past year of commit activity
  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

  •4646 forks•972972 stars•2222 issues•11 pull request•Updated Jan 5, 2026Jan 5, 2026

• vidar-robotwin

  Public
  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

  thu-ml/vidar’s past year of commit activity
  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

  thu-ml/UltraViCo.github.io’s past year of commit activity
  Project page for "UltraViCo"

  JavaScript

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

• UniCardio

  Public

  thu-ml/UniCardio’s past year of commit activity
  Versatile Cardiovascular Signal Generation with a Unified Diffusion Transformer

  Python

  •

  MIT License

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

• FrameBridge

  Public

  thu-ml/FrameBridge’s past year of commit activity
  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

  thu-ml/DiffusionBridge’s past year of commit activity
  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

  thu-ml/EmbodiedActiveDefense’s past year of commit activity
  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