Skip to content
@thu-ml

TSAIL group

Tsinghua Statistical Artificial Intelligence & Learning Group

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

    Quantized Attention achieves speedup of 2-3x and 3-5x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.

    Cuda 1.6k 113

  3. unidiffuser unidiffuser Public

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

    Python 1.4k 90

  4. prolificdreamer prolificdreamer Public

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

    Python 1.5k 46

  5. ares ares Public

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

    Python 507 87

  6. tianshou tianshou Public

    An elegant PyTorch deep reinforcement learning library.

    Python 8.5k 1.2k

Repositories

Showing 10 of 74 repositories