Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization
☆34Oct 31, 2025Updated 4 months ago
Alternatives and similar repositories for diem
Users that are interested in diem are comparing it to the libraries listed below
Sorting:
- Official implementation of Score-based Data Assimilation☆87Jan 12, 2024Updated 2 years ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- Published in IEEE Trans. Comput. Imag.☆10Oct 22, 2024Updated last year
- Implementation of papers in 101 lines of code.☆18Nov 12, 2023Updated 2 years ago
- Pytorch wrappers for the FINUFFT library☆16Nov 21, 2025Updated 3 months ago
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆56Oct 8, 2025Updated 5 months ago
- Diffusion models in PyTorch☆128Feb 2, 2026Updated last month
- Score-Based Priors for Bayesian Inverse Imaging (ICCV 2023, TMLR 2024)☆42Dec 18, 2024Updated last year
- ☆10Apr 2, 2024Updated last year
- Repository for Angus H Wright's AstroStatistics Lecture Materials☆15Apr 6, 2022Updated 3 years ago
- Code for the publication "Appa: Bending Weather Dynamics with Latent Diffusion Models for Global Data Assimilation".☆24Dec 4, 2025Updated 3 months ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated last year
- simple simulations for testing weak lensing shear measurement☆12Oct 31, 2025Updated 4 months ago
- Lectures on Bayesian statistics and information theory☆31May 26, 2025Updated 9 months ago
- A repository to store state of the art score model architectures☆61Dec 10, 2025Updated 2 months ago
- Material for the CMU Quarks2Cosmos Conference Data Challenges☆14Aug 16, 2023Updated 2 years ago
- Coverage tests to check the quality of your posterior estimators.☆33Aug 12, 2024Updated last year
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆13Jun 24, 2023Updated 2 years ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Unleash the true power of scheduling☆33Feb 23, 2026Updated 2 weeks ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- Official implementation of the paper "Normalization-Equivariant Neural Networks with Application to Image Denoising" (NeurIPS'23)☆13Jun 27, 2025Updated 8 months ago
- A JAX package for weak gravitational lensing☆35Apr 30, 2024Updated last year
- Official Implementation for (ICLR 2025) Rethinking Diffusion Posterior Sampling: From Conditional Score Estimator to Maximizing a Posteri…☆21Jan 31, 2025Updated last year
- Code Implementation of AdjointDiffusion: a physics-guided and fabrication-aware optimization of photonic devices☆23Jan 4, 2026Updated 2 months ago
- Cosmological map inference with deep learning☆15Nov 3, 2021Updated 4 years ago
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆18Jan 6, 2026Updated 2 months ago
- Official PyTorch implementation for Decomposed Diffusion Sampling (DDS), presented in the paper "Decomposed Diffusion Sampler for Acceler…☆76Sep 26, 2024Updated last year
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆79Jun 20, 2025Updated 8 months ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆24Nov 13, 2025Updated 3 months ago
- We derive a fundamental property of the posterior distribution in Gaussian denoising, and use it to propose a new way for uncertainty vis…☆19Apr 30, 2024Updated last year
- Equivariant plug-and-play algorithms☆20Aug 26, 2025Updated 6 months ago
- ☆20Aug 30, 2023Updated 2 years ago
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Apr 26, 2024Updated last year
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆202Feb 19, 2025Updated last year
- ☆21May 14, 2025Updated 9 months ago