Alternatives and similar repositories for awesome-JAXtronomy

Users that are interested in awesome-JAXtronomy are comparing it to the libraries listed below

• dpiras / cosmopower-jax

  View on GitHub
  Differentiable cosmological emulators: the JAX version of CosmoPower
  ☆43Nov 27, 2025Updated 2 months ago
• GalacticDynamics / unxt

  View on GitHub
  Unitful Quantities in JAX
  ☆52Updated this week
• LSSTDESC / surveycodex

  View on GitHub
  The galaxy survey codex
  ☆17Jul 12, 2024Updated last year
• dfm / emcee-jax

  View on GitHub
  An experiment: emcee implemented in JAX
  ☆28Jul 1, 2022Updated 3 years ago
• exoplanet-dev / jaxoplanet

  View on GitHub
  Astronomical time series analysis with JAX
  ☆74Jan 26, 2026Updated 2 weeks ago
• johannesulf / nautilus

  View on GitHub
  Neural Network-Boosted Importance Nested Sampling for Bayesian Statistics
  ☆96Dec 29, 2025Updated last month
• benediktdiemer / galaxies_course

  View on GitHub
  Interactive notebooks for graduate course in galaxies.
  ☆16Jun 14, 2024Updated last year
• JAXtronomy / October2024-Meeting

  View on GitHub
  Astronomy Research + JAX Meeting 2024
  ☆13Oct 23, 2024Updated last year
• f0uriest / interpax

  View on GitHub
  Interpolation and function approximation with JAX
  ☆248Feb 2, 2026Updated last week
• GalacticDynamics / coordinax

  View on GitHub
  Coordinates in JAX
  ☆29Jan 31, 2026Updated 2 weeks ago
• wangbingjie / sbi_pp

  View on GitHub
  Simulation-based (likelihood-free) inference customized for astronomical applications
  ☆29Nov 4, 2024Updated last year
• kmzzhang / nbi

  View on GitHub
  Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications
  ☆39Aug 12, 2025Updated 6 months ago
• minaskar / pocomc

  View on GitHub
  pocoMC: A Python implementation of Preconditioned Monte Carlo for accelerated Bayesian Computation
  ☆118Sep 11, 2025Updated 5 months ago
• JAXtronomy / spexial

  View on GitHub
  Special functions in Jax
  ☆16Mar 1, 2025Updated 11 months ago
• nuclear-multimessenger-astronomy / jester

  View on GitHub
  JAX-accelerated nuclear equation of state inference and TOV solvers
  ☆10Feb 7, 2026Updated last week
• kazewong / flowMC

  View on GitHub
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆255Oct 28, 2025Updated 3 months ago
• GalacticDynamics / galax

  View on GitHub
  Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
  ☆41Jan 31, 2026Updated 2 weeks ago
• tedwards2412 / ripple

  View on GitHub
  Differentiable Gravitational Waveforms with JAX
  ☆71Jul 28, 2025Updated 6 months ago
• ChrisBoettner / gallifrey

  View on GitHub
  A JAX-based gaussian process structure learning package for time series modelling.
  ☆34May 28, 2025Updated 8 months ago
• astrockragh / Mangrove

  View on GitHub
  Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.
  ☆24Jul 24, 2025Updated 6 months ago
• avapolzin / spike

  View on GitHub
  All-in-one tool to generate, and correctly drizzle, HST, JWST, and Roman PSFs.
  ☆13Jan 26, 2026Updated 2 weeks ago
• ArgonneCPAC / dsps

  View on GitHub
  Differentiable Stellar Population Synthesis
  ☆32Jan 19, 2026Updated 3 weeks ago
• TomWagg / software-citation-station

  View on GitHub
  A website for making citing software used in your research quick and easy
  ☆29Updated this week
• JohannesBuchner / UltraNest

  View on GitHub
  Fit and compare complex models reliably and rapidly. Advanced nested sampling.
  ☆197Jan 22, 2026Updated 3 weeks ago
• EiffL / LLMAssistedResearch

  View on GitHub
  Exploring how ChatGPT can be used to accelerate research in cosmology.
  ☆12Dec 12, 2022Updated 3 years ago
• DifferentiableUniverseInitiative / GalFlow

  View on GitHub
  A TensorFlow reimplementation of GalSim
  ☆10Mar 8, 2022Updated 3 years ago
• jwuphysics / galaxy-autoencoders

  View on GitHub
  Autoencoders for galaxies
  ☆11Jul 29, 2021Updated 4 years ago
• lenstronomy / JAXtronomy

  View on GitHub
  JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.
  ☆19Jan 23, 2026Updated 3 weeks ago
• Joshuaalbert / jaxns

  View on GitHub
  Probabilistic Programming and Nested sampling in JAX
  ☆221Dec 15, 2025Updated last month
• tousifislam / gwModels

  View on GitHub
  Package for data-driven and phenomenological gravitational waveform models
  ☆12Feb 12, 2025Updated last year
• DifferentiableUniverseInitiative / jax_cosmo

  View on GitHub
  A differentiable cosmology library in JAX
  ☆223Jun 27, 2025Updated 7 months ago
• pmocz / jax-euler-benchmarks

  View on GitHub
  A simple JAX solver for the 2D Euler equations for benchmarking
  ☆19Nov 3, 2025Updated 3 months ago
• changhoonhahn / SEDflow

  View on GitHub
  Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation
  ☆25Dec 11, 2025Updated 2 months ago
• astro-informatics / proxnest

  View on GitHub
  Proximal Nested Sampling for high-dimensional Bayesian model selection
  ☆26Apr 18, 2024Updated last year
• undark-lab / swyft

  View on GitHub
  A system for scientific simulation-based inference at scale.
  ☆165Mar 30, 2024Updated last year
• kartheikiyer / GP-SFH

  View on GitHub
  Gaussian process based implementation for galaxy star formation histories with custom kernels
  ☆16Oct 10, 2025Updated 4 months ago
• JakobRobnik / MicroCanonicalHMC

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

  View on GitHub
  ☆216Feb 1, 2026Updated last week
• NIFTy-PPL / JAXbind

  View on GitHub
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆75Updated this week