Alternatives and similar repositories for fab-jax

Users that are interested in fab-jax are comparing it to the libraries listed below

• balintmate / boltzmann-interpolations

  View on GitHub
  ☆11Oct 15, 2023Updated 2 years ago
• lollcat / fab-torch

  View on GitHub
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆66Mar 11, 2024Updated last year
• albcab / mfm

  View on GitHub
  Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows
  ☆13May 23, 2024Updated last year
• kilian-group / LMLM

  View on GitHub
  ☆23Oct 31, 2025Updated 3 months ago
• google-deepmind / annealed_flow_transport

  View on GitHub
  ☆52Feb 27, 2023Updated 2 years ago
• shreyaspadhy / CMCD

  View on GitHub
  Implementing Controlled Monte Carlo Diffusions (ICLR 2024)
  ☆17Sep 30, 2024Updated last year
• DenisBless / variational_sampling_methods

  View on GitHub
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆46Dec 9, 2024Updated last year
• katalinic / sdflows

  View on GitHub
  JAX exponential map normalising flows on sphere
  ☆17Oct 4, 2020Updated 5 years ago
• necludov / action-matching

  View on GitHub
  Implementation of Action Matching for the Schrödinger equation
  ☆25Jun 18, 2023Updated 2 years ago
• jarridrb / DEM

  View on GitHub
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆68Apr 11, 2025Updated 10 months ago
• bc / retina

  View on GitHub
  Tools for visualizing and comparing data from vertebrate retinas
  ☆14Jan 20, 2025Updated last year
• zdhNarsil / Diffusion-Generative-Flow-Samplers

  View on GitHub
  PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…
  ☆26Dec 21, 2023Updated 2 years ago
• dgerosa / astrostatistics_bicocca_2025

  View on GitHub
  Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)
  ☆15Aug 30, 2025Updated 5 months ago
• modichirag / GSM-VI

  View on GitHub
  Code for Gaussian Score Matching Variational Inference
  ☆35Feb 23, 2025Updated 11 months ago
• JakobRobnik / MicroCanonicalHMC

  View on GitHub
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆75Dec 9, 2025Updated 2 months ago
• tumaer / neuralsph

  View on GitHub
  Neural SPH
  ☆41Nov 23, 2025Updated 2 months ago
• noame12 / Explainable_Attention_Based_Deepfake_Detector

  View on GitHub
  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
• rmcgibbo / pyhmc

  View on GitHub
  Hamiltonain Monte Carlo in Python
  ☆40Nov 7, 2019Updated 6 years ago
• cg-gdsc / GDSC-6

  View on GitHub
  This is the official GDSC repo with all of the source code presented in the video tutorials
  ☆14Jun 27, 2023Updated 2 years ago
• matejgrcic / DenseFlow

  View on GitHub
  Official implementation of Densely connected normalizing flows
  ☆37Dec 17, 2023Updated 2 years ago
• stephengreen / gw-school-corfu-2023

  View on GitHub
  Tutorial for GW parameter inference with machine learning
  ☆12Sep 30, 2023Updated 2 years ago
• Yancccccc / HyFormer

  View on GitHub
  HyFormer: Hybrid Transformer and CNN For Pixel-level Multispectral Image Classification
  ☆15Feb 15, 2023Updated 3 years ago
• Jiaming-Liu / awesome-image-denoising-state-of-the-art

  View on GitHub
  awesome image and video denoising, state of the art networks
  ☆10Aug 2, 2019Updated 6 years ago
• luciapapalini / kenn-gw-transformer

  View on GitHub
  KENN - A Transformer Encoder for Gravitational Waves
  ☆11Sep 29, 2025Updated 4 months ago
• VTDA-Group / superphot-plus

  View on GitHub
  Superphot+, upgraded version of Superphot that uses nested sampling
  ☆12Dec 31, 2025Updated last month
• zhu-xlab / ML4Earth-Hackathon-2022

  View on GitHub
  ☆13Dec 6, 2022Updated 3 years ago
• Baran-phys / Tropical-Attention

  View on GitHub
  [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"
  ☆23Oct 23, 2025Updated 3 months ago
• OllieBoyne / sslap

  View on GitHub
  Auction Algorithm for Sparse Linear Assignment Problems
  ☆13Mar 15, 2021Updated 4 years ago
• HwijaeSon / AL-PINNs

  View on GitHub
  Official code for AL-PINNS: Augmented Lagrangian relaxation method for Physics-Informed Neural Networks
  ☆12Jul 29, 2023Updated 2 years ago
• bmad-sim / Bmad-X

  View on GitHub
  Experimental Bmad Code transcribed in Python with Numba and Pytorch
  ☆14Jun 21, 2024Updated last year
• zhwm / SparsePro

  View on GitHub
  A fine-mapping method integrating GWAS summary statistics and functional annotation data
  ☆11Dec 28, 2023Updated 2 years ago
• gregreen / deep-potential

  View on GitHub
  Deep learning for gravitational potentials, based on well-mixed tracers in phase space.
  ☆11Dec 29, 2025Updated last month
• bhagya-hettige / GLACE

  View on GitHub
  Gaussian Embedding of Large-scale Attributed Graphs
  ☆10Mar 13, 2020Updated 5 years ago
• daviesl / trjp

  View on GitHub
  Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.
  ☆10Mar 3, 2023Updated 2 years ago
• Plasmacaster / Optimized_SavGol

  View on GitHub
  Self-optimizing nth order Savitzky-Golay filter
  ☆12Aug 31, 2023Updated 2 years ago
• floapfel / MAMBA-implementations

  View on GitHub
  Implementation of a simple linear regression algorithm in MAMBA
  ☆10Feb 12, 2020Updated 6 years ago
• nuclear-multimessenger-astronomy / jester

  View on GitHub
  JAX-accelerated nuclear equation of state inference and TOV solvers
  ☆10Updated this week
• TimeDevBlocker / TransientViT

  View on GitHub
  TransientViT: A novel CNN - Vision Transformer hybrid real/bogus transient classifier for the Kilodegree Automatic Transient Survey
  ☆10Nov 7, 2024Updated last year
• GiacomoRicigliano / xkn

  View on GitHub
  Framework for computing kilonova light curves
  ☆12Sep 2, 2025Updated 5 months ago