Using Gaussian Analytical Solution to accelerate diffusion sampling. Official code repo for the paper TMLR 2024, "The Unreasonable Effectiveness of Gaussian Score Approximation for Diffusion Models…

Deep universal probabilistic programming with Python and PyTorch

[ICLR 2025] Cheating Automatic LLM Benchmarks: Null Models Achieve High Win Rates (Oral)

Bayesian Simultaneous Spectroscopic Fitting

ATP Tennis Rankings, Results, and Stats

attention is all you need!

The AI Code Editor

TorchCFM: a Conditional Flow Matching library

Tutorial Jupyter Notebooks for Data Preview 0, created and maintained by the Rubin Observatory Community Science Team.

Implementation of Real NVP in PyTorch

Posterior with interesting shapes from actually used models

Codebase for Paper Reusing Embeddings: Reproducible Reward Model Research in Large Language Model Alignment without GPUs

SSD: A Unified Framework for Self-Supervised Outlier Detection [ICLR 2021]

Code for the Interspeech 2021 paper "AST: Audio Spectrogram Transformer".

Official implementation of Diffusion Autoencoders

Linux running inside a PDF file via a RISC-V emulator

Fully open reproduction of DeepSeek-R1

Python library for supernova cosmology

cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffDock, MACE, Allegro and NEQUIP, based on equivariant neural n…

Reusable BatchBALD implementation

A Computational Fluid Dynamics (CFD) course with Python

An individual-based model that predicts the movement paths of animals by incorporating empirical distributions of fundamental step lengths (Fstepdistance), turn angles (Fturnangle), and compass hea…

Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding

Source code for ICLR 2018 Paper: Active Learning for Convolutional Neural Networks: A Core-Set Approach

Deep Active Learning