tonyduan/mixture-density-network

Normalizing flows in PyTorch

Implementation of rectified flow and some of its followup research / improvements in Pytorch

Karras et al. (2022) diffusion models for PyTorch

Open-source framework for uncertainty and deep learning models in PyTorch 🌱

Uncertainty quantification with PyTorch

Deep universal probabilistic programming with Python and PyTorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

A highly efficient implementation of Gaussian Processes in PyTorch

Lightning-UQ-Box: Uncertainty Quantification for Neural Networks with PyTorch and Lightning