Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
☆13Jun 12, 2024Updated last year
Alternatives and similar repositories for fab-jax
Users that are interested in fab-jax are comparing it to the libraries listed below
Sorting:
- ☆11Oct 15, 2023Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆67Mar 11, 2024Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- ☆52Feb 27, 2023Updated 3 years ago
- ☆25Oct 31, 2025Updated 4 months ago
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆17Sep 30, 2024Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Dec 9, 2024Updated last year
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆68Apr 11, 2025Updated 10 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Dec 21, 2023Updated 2 years ago
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆15Aug 30, 2025Updated 6 months ago
- Tools for visualizing and comparing data from vertebrate retinas☆14Jan 20, 2025Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Feb 23, 2025Updated last year
- MCHMC: sampler from an arbitrary differentiable distribution☆75Dec 9, 2025Updated 2 months ago
- Neural SPH☆41Nov 23, 2025Updated 3 months ago
- A Deepfake detector based on hybrid EfficientNet CNN and Vision Transformer archietcture. The model is explainable by rendering a heatma…☆15Mar 16, 2022Updated 3 years ago
- Hamiltonain Monte Carlo in Python☆40Nov 7, 2019Updated 6 years ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks☆12Jul 29, 2023Updated 2 years ago
- awesome image and video denoising, state of the art networks☆10Aug 2, 2019Updated 6 years ago
- Framework for computing kilonova light curves☆12Sep 2, 2025Updated 6 months ago
- Superphot+, upgraded version of Superphot that uses nested sampling☆12Dec 31, 2025Updated 2 months ago
- Tutorial for GW parameter inference with machine learning☆12Sep 30, 2023Updated 2 years ago
- KENN - A Transformer Encoder for Gravitational Waves☆11Sep 29, 2025Updated 5 months ago
- ☆13Dec 6, 2022Updated 3 years ago
- Gaussian Embedding of Large-scale Attributed Graphs☆10Mar 13, 2020Updated 5 years ago
- Official implementation of Densely connected normalizing flows☆37Dec 17, 2023Updated 2 years ago
- TransientViT: A novel CNN - Vision Transformer hybrid real/bogus transient classifier for the Kilodegree Automatic Transient Survey☆10Nov 7, 2024Updated last year
- Implementation of a simple linear regression algorithm in MAMBA☆10Feb 12, 2020Updated 6 years ago
- HyFormer: Hybrid Transformer and CNN For Pixel-level Multispectral Image Classification☆16Feb 15, 2023Updated 3 years ago
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆23Oct 23, 2025Updated 4 months ago
- Experimental Bmad Code transcribed in Python with Numba and Pytorch☆14Jun 21, 2024Updated last year
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Updated this week
- This is the official GDSC repo with all of the source code presented in the video tutorials☆14Jun 27, 2023Updated 2 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- Deep learning for gravitational potentials, based on well-mixed tracers in phase space.☆11Dec 29, 2025Updated 2 months ago
- ☆16Feb 27, 2026Updated last week
- Auction Algorithm for Sparse Linear Assignment Problems☆13Mar 15, 2021Updated 4 years ago