Pinned Loading

1. zhusuan zhusuan Public

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

   Python 2.2k 420

2. SageAttention SageAttention Public

   [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 2.8k 274

3. unidiffuser unidiffuser Public

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

   Python 1.5k 90

4. prolificdreamer prolificdreamer Public

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

   Python 1.6k 47

5. ares ares Public

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

   Python 519 92

6. tianshou tianshou Public

   An elegant PyTorch deep reinforcement learning library.

   Python 9k 1.2k