TSAIL group ORG

An elegant PyTorch deep reinforcement learning library.

Tianshou is a Python reinforcement learning library built on PyTorch that provides both high-level training APIs and low-level algorithm customization for researchers and practitioners.

TurboDiffusion: 100–200× Acceleration for Video Diffusion Models

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow

RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation

ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation (NeurIPS 2023 Spotlight)

Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"

Official code of Motus: A Unified Latent Action World Model

[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.

Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) , UltraViCo (ICLR 2026) and UltraImage

[ICML 2026] Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation" & Causal Forcing++

Official code of RDT 2

[ECCV 2024] Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.

A Python library for adversarial machine learning focusing on benchmarking adversarial robustness.

SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention

Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)

Official implementation for "ControlVideo: Adding Conditional Control for One Shot Text-to-Video Editing"

A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)

Benchmarking Robustness of 3D Object Detection to Common Corruptions in Autonomous Driving, CVPR 2023

Cache efficient implementation for Latent Dirichlet Allocation

Low-bit optimizers for PyTorch

Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)

Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).

Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)

[ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.

No description.

天授中文文档

Hierarchical Decomposition of Prompt-Based Continual Learning: Rethinking Obscured Sub-optimality (NeurIPS 2023, Spotlight)

No description.

Official codebase for "STAIR: Improving Safety Alignment with Introspective Reasoning"

Stochastic training of graph convolutional networks

No description.

A toolbox for benchmarking Multimodal LLM Agents trustworthiness across truthfulness, controllability, safety and privacy dimensions through 34 interactive tasks

No description.

Versatile Cardiovascular Signal Generation with a Unified Diffusion Transformer

Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)

Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"

No description.

Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior" (ICLR 2024).

Towards Effective Adversarial Textured 3D Meshes on Physical Face Recognition, CVPR 2023, Highlight

Official repo for vidar and vidarc: video foundation model for robotics.

No description.

Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training

Official code for "VFlow: More Expressive Generative Flows with Variational Data Augmentation" (ICML 2020)

Codes accompanying the paper "Toward Guidance-Free AR Visual Generation via Condition Contrastive Alignment"

Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)

Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction

[ICML 2023] Non-Uniform Neural Operator (NUNO)

Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"

Big Topic Model is a fast engine for running large-scale Topic Models.

A toolkit for processing raw embodied data into standardized formats and converting between embodied dataset schemas.

No description.

Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)

Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)

Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)

Code for "Function Space Particle Optimization for Bayesian Neural Networks"

IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)

No description.

Code for "Collaborative Filtering with User-Item Co-Autoregressive Models"

Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)

Zhusuan with backend Jittor

[ICML 2026 Spotlight] Official implementation of TetraJet-v2: Accurate NVFP4 Training for LLMs, with fully-NVFP4 linear layer with unbiased backprop, and algorithms to overcome LLMs' weight-oscillation and activation-outlier bottlenecks.

No description.

mmDCGMs for accurate classification and excellent class-conditional generation in semi-supervised learning

Official implementation for "Reinforced Embodied Active Defense: Exploiting Adaptive Interaction for Robust Visual Perception in Adversarial 3D Environments" (TPAMI 2025)

Official code for "Efficient Backpropagation with Variance Controlled Adaptive Sampling" (ICLR 2024)

No description.

Zhusuan with backend PaddlePaddle

Code for "Learning to Generate with Memory"

No description.

No description.

Population matching discrepancy

[ICML 2026 Spotlight] Official implementation of "PACT: Self-Evolving Physical Safety Alignment for Diffusion Policies in Embodied Manipulation".

robotwin evaluation code for vidar.

code for pLSA and LDA in the paper "Stochastic Expectation Maximization with Variance Reduction"

No description.

No description.

No description.

FrameBridge: Improving Image-to-Video Generation with Bridge Models

Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.

Code for "A Wasserstein Minimum Velocity Approach to Learning Unnormalized Models"

No description.

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model"

No description.

No description.

No description.

Project page for "UltraViCo"

No description.

No description.

No description.

conda install tianshou -c conda-forge

Mirror of Apache Spark