Diffusion Model Python Packages

avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

smalldiffusion

Simple and readable code for training and sampling from diffusion models

molcraftdiffusion

A generative AI framework for 3D molecular generation and data-driven molecular design in computational chemistry.

diffusers-dmse

PyPI package for MSE-optimal denoising with diffusion models.

multicom-ligand

Comprehensive ensembling of protein-ligand structure and affinity prediction methods (CASP16)