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 lossing end-to-end metrics across language, image, and video models.

   Cuda 2.6k 244

3. unidiffuser unidiffuser Public

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

   Python 1.4k 88

4. prolificdreamer prolificdreamer Public

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

   Python 1.5k 47

5. ares ares Public

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

   Python 516 89

6. tianshou tianshou Public

   An elegant PyTorch deep reinforcement learning library.

   Python 8.9k 1.2k