TSAIL group

All

77 repositories

SageAttention
Public
Quantized Attention achieves speedup of 2-5x and 3-11x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.
cuda triton attention vit quantization video-generation mlsys inference-acceleration efficient-attention llm
Cuda
•
Apache License 2.0
•139•1.8k•58•9•Updated Jul 1, 2025Jul 1, 2025
MLA-Trust
Public
A toolbox for benchmarking Multimodal LLM Agents trustworthiness across truthfulness, controllability, safety and privacy dimensions through 34 interactive tasks
agent benchmark privacy toolbox safety multi-modal controllability trustworthy-ai truthfulness
Python
•
MIT License
•3•29•0•0•Updated Jun 30, 2025Jun 30, 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.2k•8.6k•145•8•Updated Jun 27, 2025Jun 27, 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
•10•149•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
•1•16•0•0•Updated Jun 20, 2025Jun 20, 2025
SpargeAttn
Public
SpargeAttention: A training-free sparse attention that can accelerate any model inference.
attention vit quantization video-generation mlsys inference-acceleration ai-infra vision-transformer sparse-attention llm
Cuda
•
Apache License 2.0
•45•627•22•4•Updated Jun 19, 2025Jun 19, 2025
DDO
Public
Python
•
Other
•0•0•0•0•Updated Jun 16, 2025Jun 16, 2025
GFT
Public
Python
•
MIT License
•0•35•4•0•Updated Jun 13, 2025Jun 13, 2025
SparseDM
Public
Python
•0•1•0•0•Updated Jun 10, 2025Jun 10, 2025
RoboticsDiffusionTransformer
Public
RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation
Python
•
MIT License
•128•1.3k•25•5•Updated Jun 5, 2025Jun 5, 2025
FrameBridge
Public
FrameBridge: Improving Image-to-Video Generation with Bridge Models
0•0•0•0•Updated May 26, 2025May 26, 2025
UniCardio
Public
This is the official code of UniCardio
0•6•0•0•Updated May 25, 2025May 25, 2025
RIFLEx
Public
Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025)
generative-model rope diffusion extrapolation dit video-generation position-embedding diffusion-models long-video-generation diffusion-transformer
Python
•
Apache License 2.0
•67•688•20•0•Updated May 13, 2025May 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•254•8•0•Updated Apr 25, 2025Apr 25, 2025
DiffusionBridge
Public
Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)
Python
•5•67•4•0•Updated Mar 10, 2025Mar 10, 2025
i-DODE
Public
Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)
Python
•
Apache License 2.0
•1•18•1•0•Updated Mar 4, 2025Mar 4, 2025
STAIR
Public
Official codebase for "STAIR: Improving Safety Alignment with Introspective Reasoning"
Python
•
MIT License
•0•50•0•0•Updated Feb 26, 2025Feb 26, 2025
EffWRN-paddle
Public
Python
•
MIT License
•0•0•0•0•Updated Feb 24, 2025Feb 24, 2025
RIFLEx.github.io
Public
JavaScript
•0•0•0•0•Updated Feb 21, 2025Feb 21, 2025
oddefense
Public
Jupyter Notebook
•
MIT License
•1•3•0•0•Updated Feb 14, 2025Feb 14, 2025
CCA
Public
Codes accompanying the paper "Toward Guidance-Free AR Visual Generation via Condition Contrastive Alignment"
Python
•
MIT License
•0•33•0•0•Updated Feb 11, 2025Feb 11, 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
•1•12•1•0•Updated Dec 22, 2024Dec 22, 2024
ReMoE
Public
[ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.
Python
•
Other
•3•83•1•0•Updated Dec 20, 2024Dec 20, 2024
tianshou-docs-zh_CN
Public
天授中文文档
TeX
•11•58•0•0•Updated Dec 10, 2024Dec 10, 2024
2by4-pretrain-acc-examples
Public
Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•0•24•1•0•Updated Dec 8, 2024Dec 8, 2024
CRM
Public
[ECCV 2024] Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.
generative-model reconstruction 3d multiview diffusion-models aigc
Python
•
MIT License
•54•654•25•1•Updated Nov 28, 2024Nov 28, 2024
HiDe-PET
Public
IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)
Python
•
MIT License
•3•7•0•0•Updated Nov 15, 2024Nov 15, 2024
HiDe-Prompt
Public
Hierarchical Decomposition of Prompt-Based Continual Learning: Rethinking Obscured Sub-optimality (NeurIPS 2023, Spotlight)
Python
•
MIT License
•10•86•3•0•Updated Nov 15, 2024Nov 15, 2024
Noise-Contrastive-Alignment
Public
Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)
Python
•
MIT License
•5•54•2•0•Updated Nov 8, 2024Nov 8, 2024
Efficient-Diffusion-Alignment
Public
Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)
Python
•0•13•1•0•Updated Oct 29, 2024Oct 29, 2024