TSAIL group

All

89 repositories

SpargeAttn
Public
[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.
attention vit quantization video-generation mlsys inference-acceleration ai-infra vision-transformer sparse-attention llm
Cuda
•
Apache License 2.0
•87•952•56•2•Updated Feb 25, 2026Feb 25, 2026
embodied-data-toolkit
Public
A toolkit for processing raw embodied data into standardized formats and converting between embodied dataset schemas.
data-processing embodied-ai format-converter-tool
Python
•2•15•0•0•Updated Feb 24, 2026Feb 24, 2026
SLA
Public
SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention
transformer video-generation mlsys inference-acceleration ai-infra linear-attention sparse-attention diffusion-transformer train-acceleration sparse-linear-attention
Python
•
Apache License 2.0
•17•282•7•0•Updated Feb 24, 2026Feb 24, 2026
Causal-Forcing
Public
Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation"
diffusion distillation video-generation world-models diffusion-models consistency-models text-to-video autoregressive-models generative-ai world-model
Python
•
Apache License 2.0
•18•395•8•0•Updated Feb 23, 2026Feb 23, 2026
TurboDiffusion
Public
TurboDiffusion: 100–200× Acceleration for Video Diffusion Models
rcm distillation video-generation mlsystem diffusion-models inference-acceleration ai-infra consistency-model sageattention sparse-linear-attention
Python
•
Apache License 2.0
•233•3.4k•61•8•Updated Feb 20, 2026Feb 20, 2026
CausalForcing.github.io
Public
JavaScript
•0•0•0•0•Updated Feb 7, 2026Feb 7, 2026
RDT2
Public
Official code of RDT 2
Python
•
Apache License 2.0
•43•725•14•2•Updated Feb 7, 2026Feb 7, 2026
ultraimage.github.io
Public
JavaScript
•0•0•0•0•Updated Feb 6, 2026Feb 6, 2026
DiT-Extrapolation
Public
Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) and UltraViCo (ICLR 2026)
generative-model rope diffusion extrapolation dit video-generation position-embedding diffusion-models long-video-generation diffusion-transformer
Python
•
Apache License 2.0
•74•785•23•0•Updated Feb 2, 2026Feb 2, 2026
i-DODE
Public
Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)
Python
•
Apache License 2.0
•1•19•1•0•Updated Jan 27, 2026Jan 27, 2026
RoboticsDiffusionTransformer
Public
RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation
Python
•
MIT License
•151•1.6k•37•3•Updated Jan 21, 2026Jan 21, 2026
SageAttention
Public
[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across la…
cuda triton attention vit quantization video-generation mlsys inference-acceleration efficient-attention llm
Cuda
•
Apache License 2.0
•360•3.2k•151•19•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…
agent benchmark privacy toolbox safety multi-modal controllability trustworthy-ai truthfulness
Python
•
MIT License
•4•64•2•0•Updated Jan 9, 2026Jan 9, 2026
Motus
Public
Official code of Motus: A Unified Latent Action World Model
robotics diffusion-model video-generation robotic-manipulation world-model unidiffuser vision-language-action-model
Python
•
Apache License 2.0
•30•796•21•1•Updated Jan 5, 2026Jan 5, 2026
vidar-robotwin
Public
robotwin evaluation code for vidar.
Python
•
MIT License
•0•4•1•0•Updated Dec 22, 2025Dec 22, 2025
vidar
Public
Official repo for vidar and vidarc: video foundation model for robotics.
Python
•1•37•1•0•Updated Dec 22, 2025Dec 22, 2025
UltraViCo.github.io
Public
Project page for "UltraViCo"
JavaScript
•0•0•0•0•Updated Dec 3, 2025Dec 3, 2025
tianshou
Public
An elegant PyTorch deep reinforcement learning library.
pytorch dqn policy-gradient rl cql atari ddpg imitation-learning sac drl
Python
•
MIT License
•1.3k•10k•132•2•Updated Dec 1, 2025Dec 1, 2025
UniCardio
Public
Versatile Cardiovascular Signal Generation with a Unified Diffusion Transformer
Python
•
MIT License
•3•50•2•0•Updated Oct 9, 2025Oct 9, 2025
FrameBridge
Public
FrameBridge: Improving Image-to-Video Generation with Bridge Models
Python
•0•3•0•0•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
•10•107•6•0•Updated Sep 9, 2025Sep 9, 2025
SparseDM
Public
Python
•1•1•0•0•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
•0•8•0•0•Updated Jul 26, 2025Jul 26, 2025
oddefense
Public
Jupyter Notebook
•
MIT License
•2•5•0•0•Updated Jul 16, 2025Jul 16, 2025
Adaptive-Sparse-Trainer
Public
Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)
Python
•
Apache License 2.0
•2•18•0•0•Updated Jul 1, 2025Jul 1, 2025
MMTrustEval
Public
A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)
benchmark privacy toolbox safety multi-modal fairness robustness claude gpt-4 trustworthy-ai
Python
•
Creative Commons Attribution Share Alike 4.0 International
•12•174•1•0•Updated Jun 27, 2025Jun 27, 2025
TetraJet-MXFP4Training
Public
Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training
Python
•
Apache License 2.0
•4•36•0•0•Updated Jun 20, 2025Jun 20, 2025
DDO
Public
Python
•
Other
•0•0•0•0•Updated Jun 16, 2025Jun 16, 2025
GFT
Public
Python
•
MIT License
•1•52•5•0•Updated Jun 13, 2025Jun 13, 2025
cond-image-leakage
Public
Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)
Apache License 2.0
•22•258•8•0•Updated Apr 25, 2025Apr 25, 2025