Alternatives and similar repositories for exojax

Users that are interested in exojax are comparing it to the libraries listed below

• astro-informatics / proxnest
  Proximal Nested Sampling for high-dimensional Bayesian model selection
  ☆24Updated last year
• dfm / numpyro-ext
  A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work
  ☆24Updated 4 months ago
• JAXtronomy / October2024-Meeting
  Astronomy Research + JAX Meeting 2024
  ☆13Updated 7 months ago
• exoplanet-dev / jaxoplanet
  Astronomical time series analysis with JAX
  ☆60Updated this week
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆40Updated this week
• tedwards2412 / ripple
  Differentiable Gravitational Waveforms with JAX
  ☆57Updated 9 months ago
• JAXtronomy / awesome-JAXtronomy
  Curated list of astronomy/astrophysics code packages using JAX
  ☆43Updated 6 months ago
• DifferentiableUniverseInitiative / JaxPM
  JAX-powered Cosmological Particle-Mesh N-body Solver
  ☆43Updated 3 weeks ago
• htjb / margarine
  Code to replicate posterior probability distributions with bijectors/KDEs and perform marginal KL/bayesian dimensionality calculations.
  ☆13Updated 3 months ago
• tingyuansen / deep-potential
  ☆16Updated 4 years ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆36Updated last month
• kmzzhang / nbi
  Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications
  ☆39Updated 2 months ago
• GalacticDynamics / galax
  Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
  ☆39Updated this week
• kazewong / jim
  Gravitational-wave data analysis tools in Jax
  ☆71Updated this week
• dpiras / cosmopower-jax
  Differentiable cosmological emulators: the JAX version of CosmoPower
  ☆37Updated last week
• nmoteki / ndimsplinejax
  A multidimensional spline interpolation by Google/JAX
  ☆12Updated last year
• dfm / emcee-jax
  An experiment: emcee implemented in JAX
  ☆25Updated 2 years ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆72Updated last month
• SebastianBocquet / pygtc
  Make a sweet giant triangle confusogram (GTC) plot
  ☆34Updated last year
• h3jia / bayesfast
  Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.
  ☆27Updated 2 years ago
• ColmTalbot / gwpopulation
  CPU/GPU agnostic gravitational-wave population inference
  ☆47Updated this week
• eelregit / pmwd
  Differentiable Cosmological Forward Model
  ☆79Updated 3 months ago
• homerjed / sbiax
  Fast, lightweight and parallelised simulation-based inference in JAX.
  ☆18Updated 2 weeks ago
• tomcharnock / IMNN
  Using neural networks to extract sufficient statistics from data by maximising the Fisher information
  ☆32Updated 2 years ago
• johannesulf / nautilus
  Neural Network-Boosted Importance Nested Sampling for Bayesian Statistics
  ☆86Updated 2 weeks ago
• jecampagne / JaxTutos
  JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
  ☆15Updated 3 months ago
• glass-dev / glass
  Generator for Large Scale Structure
  ☆39Updated this week
• smsharma / jax-conditional-flows
  Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
  ☆20Updated last year
• EiffL / JaxKDTree
  A Jax wrapper for cudaKDTree
  ☆10Updated last year
• GalacticDynamics / coordinax
  Coordinates in JAX
  ☆18Updated last week