Score Matching Python Packages

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

torchebm

🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffusion, flow matching, and Schrödinger bridges.

gensbi

Generative Models for Simulation-Based Inference in JAX

iscan-dag

A Python 3 package for identifying distribution shifts (a.k.a feature-shifts) between datasets. Official implementation of the paper: "iSCAN: Identifying Causal Mechanism Shifts among Nonlinear Additive Noise Models".

deterministicparticleflowcontrol

Repository for Deterministic Particle Flow Control framework

toy-gradlogp

Some toy examples of score matching algorithms written in PyTorch

freqdiff

This repository implements time series diffusion in the frequency domain.