Score Matching Python Packages

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

torchebm

🍓 Build and train energy-based and diffusion models in PyTorch ⚡.

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".

toy-gradlogp

Some toy examples of score matching algorithms written in PyTorch

deterministicparticleflowcontrol

DeterministicParticleFlowControl: deterministic particle based stochastic optimal control framework

freqdiff

Time series diffusion in the frequency domain.