Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
☆13Jun 12, 2024Updated 2 years ago
Alternatives and similar repositories for fab-jax
Users that are interested in fab-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆13Oct 15, 2023Updated 3 years ago
- Tutorial for paper "Flexible Gravitational-Wave Parameter Estimation with Transformers"☆16Jun 24, 2026Updated 4 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆70Mar 11, 2024Updated 2 years ago
- ☆53Feb 27, 2023Updated 3 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆11Jun 1, 2025Updated last year
- Tutorial for making figures, written by Michael @michaeldeistler☆13May 15, 2025Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- Normalising flows implemented using nflows☆27Updated this week
- Tutorial for GW parameter inference with machine learning☆13Sep 30, 2023Updated 3 years ago
- A multidimensional spline interpolation by Google/JAX☆20Feb 13, 2024Updated 2 years ago
- This repository contains the code for the paper "Can Transformers Learn Full Bayesian Inference In Context?"☆18Apr 6, 2025Updated last year
- Implementation of RigidFormer, Learning Rigid Dynamics using Transformers☆32Aug 11, 2026Updated 2 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆25Dec 21, 2023Updated 2 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 6 years ago
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆18Sep 30, 2024Updated 2 years ago
- Enjoy dark mode on Arxiv papers☆23Aug 16, 2021Updated 5 years ago
- Official repository for the preprint "On Sampling with Approximate Transport Maps" in PyTorch.☆10May 28, 2023Updated 3 years ago
- Sampling with Blackjax on Aesara☆11Mar 10, 2023Updated 3 years ago
- ☆10Mar 14, 2022Updated 4 years ago
- doi2bibtex: Resolve DOIs and arXiv identifiers to formatted BibTeX entries☆23Aug 20, 2024Updated 2 years ago
- ☆24Aug 15, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Bijections & normalizing flows with JAX/NNX☆30Aug 23, 2026Updated 2 months ago
- awesome image and video denoising, state of the art networks☆10Aug 2, 2019Updated 7 years ago
- Repository of the journal club "Diffusion Models and Generative Modeling"☆14Nov 20, 2024Updated last year
- A pytorch version of hamiltonian monte carlo☆14Jun 26, 2019Updated 7 years ago
- [NeurIPS 2023] "Natural Actor-Critic for Robust Reinforcement Learning with Function Approximation"☆12Oct 20, 2024Updated 2 years ago
- 📙 A Community Database of Accelerator Simulation Codes☆13Jan 19, 2026Updated 9 months ago
- Repository for the Summer Tutorials on Computational Physics at Ohio University☆11May 19, 2026Updated 5 months ago
- website☆12Oct 11, 2026Updated 3 weeks ago
- C++ and Python TPSA/DA codes☆10Jul 8, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Code for the paper "Learning to Do or Learning While Doing: Reinforcement Learning and Bayesian Optimisation for Online Continuous Tuning…☆14Nov 15, 2023Updated 2 years ago
- Implements Generative Phase Space Reconstruction (GPSR) which reconstructs phase space distributions from experimental data☆13Updated this week
- A fast CUDA implementation of the Linear Assignment Problem (LAP) solver for PyTorch.☆11Aug 14, 2025Updated last year
- Projected Overrelaxed Jacobi (JORProx) and Gauss-Seidel (SORProx) GPU implementations.☆14Apr 18, 2026Updated 6 months ago
- Experimental Bmad Code transcribed in Python with Numba and Pytorch☆14Jun 21, 2024Updated 2 years ago
- The first hardware-accelerated benchmark for safe RL in high-fidelity 3D physics environments.☆19Oct 28, 2026Updated last week
- Solution #5 for the kaggle competition "TrackML Particle Tracking Challenge"☆10Oct 7, 2018Updated 8 years ago