-
tianshou
An elegant PyTorch deep reinforcement learning library.
Python ★ 11k 2mo agoExplain → -
TurboDiffusion
TurboDiffusion: 100–200× Acceleration for Video Diffusion Models
Python ★ 3.5k 4d agoExplain → -
SageAttention
[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.
Cuda ★ 3.4k 5mo agoExplain → -
zhusuan
A probabilistic programming library for Bayesian deep learning, generative models, based on Tensorflow
Python ★ 2.2k 3y agoExplain → -
RoboticsDiffusionTransformer
RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation
Python ★ 1.7k 5mo agoExplain → -
prolificdreamer
ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation (NeurIPS 2023 Spotlight)
Python ★ 1.6k 2y agoExplain → -
unidiffuser
Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"
Python ★ 1.5k 3y agoExplain → -
Motus
Official code of Motus: A Unified Latent Action World Model
Python ★ 1.2k 5mo agoExplain → -
SpargeAttn
[ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.
Cuda ★ 1.0k 3mo agoExplain → -
DiT-Extrapolation
Official implementation for "RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers" (ICML 2025) , UltraViCo (ICLR 2026) and UltraImage
Python ★ 808 14d agoExplain → -
Causal-Forcing
[ICML 2026] Official codebase for "Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation" & Causal Forcing++
Python ★ 791 4d agoExplain → -
RDT2
Official code of RDT 2
Python ★ 783 4mo agoExplain → -
CRM
[ECCV 2024] Single Image to 3D Textured Mesh in 10 seconds with Convolutional Reconstruction Model.
Python ★ 688 1y agoExplain → -
ares
A Python library for adversarial machine learning focusing on benchmarking adversarial robustness.
Python ★ 524 2y agoExplain → -
SLA
SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention
Python ★ 313 3mo agoExplain → -
cond-image-leakage
Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model" (NeurIPS 2024)
★ 257 3mo agoExplain → -
controlvideo
Official implementation for "ControlVideo: Adding Conditional Control for One Shot Text-to-Video Editing"
Python ★ 231 3y agoExplain → -
MMTrustEval
A toolbox for benchmarking trustworthiness of multimodal large language models (MultiTrust, NeurIPS 2024 Track Datasets and Benchmarks)
Python ★ 176 11mo agoExplain → -
3D_Corruptions_AD
Benchmarking Robustness of 3D Object Detection to Common Corruptions in Autonomous Driving, CVPR 2023
Python ★ 173 1y agoExplain → -
warplda
Cache efficient implementation for Latent Dirichlet Allocation
C++ ★ 164 7y agoExplain → -
low-bit-optimizers
Low-bit optimizers for PyTorch
Python ★ 139 2y agoExplain → -
DPM-Solver-v3
Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)
Jupyter Notebook ★ 134 2y agoExplain → -
Bridge-TTS
Official codebase for "Schrodinger Bridges Beat Diffusion Models on Text-to-Speech Synthesis" (https://arxiv.org/abs/2312.03491).
★ 132 1y agoExplain → -
DiffusionBridge
Official codebase for "Diffusion Bridge Implicit Models" (ICLR 2025) and "Consistency Diffusion Bridge Models" (NeurIPS 2024)
Python ★ 119 9mo agoExplain → -
ReMoE
[ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.
Python ★ 116 1y agoExplain → -
Attack-Bard
No description.
Python ★ 108 2y agoExplain → -
tianshou-docs-zh_CN
天授中文文档
TeX ★ 102 1y agoExplain → -
HiDe-Prompt
Hierarchical Decomposition of Prompt-Based Continual Learning: Rethinking Obscured Sub-optimality (NeurIPS 2023, Spotlight)
Python ★ 92 1y agoExplain → -
GNOT ⑂
No description.
Python ★ 91 2y agoExplain → -
STAIR
Official codebase for "STAIR: Improving Safety Alignment with Introspective Reasoning"
Python ★ 89 1y agoExplain → -
stochastic_gcn
Stochastic training of graph convolutional networks
Python ★ 84 3y agoExplain → -
zh-clip
No description.
Python ★ 72 3y agoExplain → -
MLA-Trust
A toolbox for benchmarking Multimodal LLM Agents trustworthiness across truthfulness, controllability, safety and privacy dimensions through 34 interactive tasks
Python ★ 64 5mo agoExplain → -
Jetfire-INT8Training
No description.
Jupyter Notebook ★ 63 1y agoExplain → -
UniCardio
Versatile Cardiovascular Signal Generation with a Unified Diffusion Transformer
Python ★ 61 8mo agoExplain → -
Noise-Contrastive-Alignment
Code accompanying the paper "Noise Contrastive Alignment of Language Models with Explicit Rewards" (NeurIPS 2024)
Python ★ 58 1y agoExplain → -
DPOT ⑂
Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"
Python ★ 56 2y agoExplain → -
GFT
No description.
Python ★ 53 1y agoExplain → -
SRPO
Codes accompanying the paper "Score Regularized Policy Optimization through Diffusion Behavior" (ICLR 2024).
Python ★ 48 2y agoExplain → -
AT3D
Towards Effective Adversarial Textured 3D Meshes on Physical Face Recognition, CVPR 2023, Highlight
Python ★ 44 2y agoExplain → -
vidar
Official repo for vidar and vidarc: video foundation model for robotics.
Python ★ 42 6mo agoExplain → -
Prior-Guided-RGF
No description.
Python ★ 42 2y agoExplain → -
TetraJet-MXFP4Training
Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training
Python ★ 40 1mo agoExplain → -
vflow
Official code for "VFlow: More Expressive Generative Flows with Variational Data Augmentation" (ICML 2020)
Python ★ 40 3y agoExplain → -
CCA
Codes accompanying the paper "Toward Guidance-Free AR Visual Generation via Condition Contrastive Alignment"
Python ★ 38 1y agoExplain → -
implicit-normalizing-flows
Code for "Implicit Normalizing Flows" (ICLR 2021 spotlight)
Python ★ 37 5y agoExplain → -
CEP-energy-guided-diffusion ⑂
Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction
Python ★ 35 2y agoExplain → -
NUNO
[ICML 2023] Non-Uniform Neural Operator (NUNO)
Python ★ 27 3y agoExplain → -
2by4-pretrain-acc-examples
Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"
Jupyter Notebook ★ 27 1y agoExplain → -
BigTopicModel
Big Topic Model is a fast engine for running large-scale Topic Models.
C++ ★ 23 9y agoExplain → -
embodied-data-toolkit
A toolkit for processing raw embodied data into standardized formats and converting between embodied dataset schemas.
Python ★ 20 3mo agoExplain → -
IODF
No description.
C++ ★ 20 2y agoExplain → -
Adaptive-Sparse-Trainer
Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)
Python ★ 19 11mo agoExplain → -
CEURL
Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)
Python ★ 19 1y agoExplain → -
i-DODE
Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)
Python ★ 19 4mo agoExplain → -
fpovi
Code for "Function Space Particle Optimization for Bayesian Neural Networks"
Python ★ 18 3y agoExplain → -
HiDe-PET
IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)
Python ★ 17 1y agoExplain → -
LM-Calibration
No description.
Python ★ 17 3y agoExplain → -
CF-UIcA
Code for "Collaborative Filtering with User-Item Co-Autoregressive Models"
Python ★ 16 8y agoExplain → -
Efficient-Diffusion-Alignment
Official Codebase for "Aligning Diffusion Behaviors with Q-functions for Efficient Continuous Control" (NeurIPS 2024)
Python ★ 15 1y agoExplain → -
Zhusuan-Jittor
Zhusuan with backend Jittor
Python ★ 15 4y agoExplain → -
TetraJet-v2-NVFP4Training
[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.
Python ★ 13 1mo agoExplain → -
adversarial_training_imagenet
No description.
Python ★ 11 3y agoExplain → -
mmdcgm-ssl
mmDCGMs for accurate classification and excellent class-conditional generation in semi-supervised learning
Python ★ 11 9y agoExplain → -
EmbodiedActiveDefense
Official implementation for "Reinforced Embodied Active Defense: Exploiting Adaptive Interaction for Robust Visual Perception in Adversarial 3D Environments" (TPAMI 2025)
Python ★ 8 11mo agoExplain → -
VCAS
Official code for "Efficient Backpropagation with Variance Controlled Adaptive Sampling" (ICLR 2024)
Python ★ 8 2y agoExplain → -
ood-dgm
No description.
Python ★ 8 5y agoExplain → -
Zhusuan-PaddlePaddle
Zhusuan with backend PaddlePaddle
Python ★ 8 4y agoExplain → -
MEM_DGM
Code for "Learning to Generate with Memory"
Python ★ 8 10y agoExplain → -
oddefense
No description.
Jupyter Notebook ★ 7 11mo agoExplain → -
ProbML-book-solution
No description.
Jupyter Notebook ★ 7 2y agoExplain → -
pmd
Population matching discrepancy
Python ★ 7 3y agoExplain → -
PACT
[ICML 2026 Spotlight] Official implementation of "PACT: Self-Evolving Physical Safety Alignment for Diffusion Policies in Embodied Manipulation".
Python ★ 6 5d agoExplain → -
vidar-robotwin
robotwin evaluation code for vidar.
Python ★ 6 6mo agoExplain → -
sEM-vr
code for pLSA and LDA in the paper "Stochastic Expectation Maximization with Variance Reduction"
C++ ★ 4 7y agoExplain → -
imagenet-a-plus
No description.
Python ★ 4 2y agoExplain → -
Auto-MIA
No description.
Python ★ 3 2mo agoExplain → -
RGBT-Clothing
No description.
Python ★ 3 2mo agoExplain → -
FrameBridge
FrameBridge: Improving Image-to-Video Generation with Bridge Models
Python ★ 3 8mo agoExplain → -
PINNacle ⑂
Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.
Python ★ 3 2y agoExplain → -
wmvl
Code for "A Wasserstein Minimum Velocity Approach to Learning Unnormalized Models"
Jupyter Notebook ★ 3 6y agoExplain → -
SparseDM
No description.
Python ★ 2 10mo agoExplain → -
dpm-solver ⑂
Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)
Jupyter Notebook ★ 2 3y agoExplain → -
conditional-image-leakage
Official implementation for "Identifying and Solving Conditional Image Leakage in Image-to-Video Diffusion Model"
★ 1 2y agoExplain → -
ACTNN-PaddlePaddle
No description.
Python ★ 1 3y agoExplain → -
CausalForcing.github.io
No description.
JavaScript ★ 0 4mo agoExplain → -
ultraimage.github.io
No description.
JavaScript ★ 0 4mo agoExplain → -
UltraViCo.github.io
Project page for "UltraViCo"
JavaScript ★ 0 6mo agoExplain → -
DDO
No description.
Python ★ 0 1y agoExplain → -
EffWRN-paddle
No description.
Python ★ 0 1y agoExplain → -
RIFLEx.github.io
No description.
JavaScript ★ 0 1y agoExplain → -
tianshou-feedstock ⑂
conda install tianshou -c conda-forge
★ 0 4y agoExplain → -
spark ⑂
Mirror of Apache Spark
Scala ★ 0 9y agoExplain →
No repos match these filters.