TSAIL group

All

61 repositories

RoboticsDiffusionTransformer
Public
RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation
Python
•
MIT License
•36•414•7•1•Updated Nov 7, 2024Nov 7, 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•101•4•0•Updated Nov 5, 2024Nov 5, 2024
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
•15•362•16•1•Updated Nov 3, 2024Nov 3, 2024
Efficient-Diffusion-Alignment
Public
Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)
Python
•0•8•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•7.9k•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•287•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•573•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•10•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
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 Jun 30, 2024Jun 30, 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•33•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•99•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•85•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•18•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•482•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