TSAIL group

All

61 repositories

SageAttention
Public
Quantized Attention that achieves speedups of 2.1x and 2.7x compared to FlashAttention2 and xformers, respectively, without lossing end-to-end metrics across various models.
attention quantization inference-acceleration llm
Python
•
BSD 3-Clause "New" or "Revised" License
•16•372•12•0•Updated Nov 11, 2024Nov 11, 2024
RoboticsDiffusionTransformer
Public
RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation
Python
•
MIT License
•36•430•10•1•Updated Nov 11, 2024Nov 11, 2024
Noise-Contrastive-Alignment
Public
Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)
Python
•
MIT License
•3•32•0•0•Updated Nov 8, 2024Nov 8, 2024
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
•7•105•4•0•Updated Nov 5, 2024Nov 5, 2024
Efficient-Diffusion-Alignment
Public
Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)
Python
•0•9•0•0•Updated Oct 29, 2024Oct 29, 2024
CCA
Public
Codes accompanying the paper "Toward Guidance-Free AR Visual Generation via Condition Contrastive Alignment"
Python
•
MIT License
•0•17•0•0•Updated Oct 28, 2024Oct 28, 2024
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.1k•8k•136•3•Updated Oct 26, 2024Oct 26, 2024
DiffusionBridge
Public
Official codebase for "Diffusion Bridge Implicit Models" (https://arxiv.org/abs/2405.15885).
0•1•1•0•Updated Oct 25, 2024Oct 25, 2024
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
•28•288•4•0•Updated Oct 22, 2024Oct 22, 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
•46•574•23•0•Updated Oct 14, 2024Oct 14, 2024
CEURL
Public
Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)
reinforcement-learning unsupervised-reinforcement-learning embodied-ai cross-embodiment-reinforcement-learning neurips24
Python
•
MIT License
•0•11•0•0•Updated Oct 13, 2024Oct 13, 2024
3D_Corruptions_AD
Public
Benchmarking Robustness of 3D Object Detection to Common Corruptions in Autonomous Driving, CVPR 2023
point-cloud self-driving-car autonomous-vehicles cvpr2023
Python
•
MIT License
•15•121•7•0•Updated Oct 9, 2024Oct 9, 2024
Jetfire-INT8Training
Public
Jupyter Notebook
•3•21•1•0•Updated Jul 21, 2024Jul 21, 2024
Bridge-TTS
Public
Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).
MIT License
•1•122•4•0•Updated Jul 12, 2024Jul 12, 2024
conditional-image-leakage
Public
Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model"
Apache License 2.0
•0•0•0•0•Updated Jun 23, 2024Jun 23, 2024
DPOT
Public
Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"
Python
•4•34•0•0•Updated Jun 10, 2024Jun 10, 2024
2by4-pretrain-acc-examples
Public
Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
Python
•
BSD 3-Clause "New" or "Revised" License
•0•10•0•0•Updated May 28, 2024May 28, 2024
IODF
Public
C++
•
MIT License
•2•18•1•0•Updated Apr 17, 2024Apr 17, 2024
i-DODE
Public
Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)
Python
•
Apache License 2.0
•1•14•1•0•Updated Apr 9, 2024Apr 9, 2024
DPM-Solver-v3
Public
Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)
Jupyter Notebook
•
MIT License
•5•100•1•0•Updated Mar 25, 2024Mar 25, 2024
VCAS
Public
Official code for "Efficient Backpropagation with Variance Controlled Adaptive Sampling" (ICLR 2024)
Python
•
Apache License 2.0
•2•7•0•0•Updated Mar 8, 2024Mar 8, 2024
Attack-Bard
Public
Python
•6•86•3•0•Updated Feb 16, 2024Feb 16, 2024
SRPO
Public
Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior" (ICLR 2024).
reinforcement-learning offline rl generative diffusion score-based-models d4rl srpo behavior-regularization
Python
•
MIT License
•1•36•0•0•Updated Feb 10, 2024Feb 10, 2024
HiDe-Prompt
Public
Hierarchical Decomposition of Prompt-Based Continual Learning: Rethinking Obscured Sub-optimality (NeurIPS 2023, Spotlight)
Python
•
MIT License
•7•74•3•0•Updated Jan 22, 2024Jan 22, 2024
prolificdreamer
Public
ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation (NeurIPS 2023 Spotlight)
nerf diffusion-model stablediffusion text-to-3d dreamfusion prolificdreamer
Python
•
Apache License 2.0
•44•1.5k•19•0•Updated Nov 22, 2023Nov 22, 2023
imagenet-a-plus
Public
Python
•
MIT License
•0•4•0•0•Updated Nov 20, 2023Nov 20, 2023
CEP-energy-guided-diffusion
Public
Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction
Python
•
MIT License
•7•22•0•0•Updated Nov 3, 2023Nov 3, 2023
ares
Public
A Python library for adversarial machine learning focusing on benchmarking adversarial robustness.
nes pca bim benchmark-framework evolutionary spsa boundary distillation fgsm adversarial-attacks
Python
•
Apache License 2.0
•89•483•9•5•Updated Oct 15, 2023Oct 15, 2023
low-bit-optimizers
Public
Low-bit optimizers for PyTorch
Python
•
Apache License 2.0
•8•118•4•1•Updated Oct 9, 2023Oct 9, 2023
PINNacle
Public
Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.
Python
•38•2•0•0•Updated Sep 20, 2023Sep 20, 2023