GalacticDynamics / unxt
Unitful Quantities in JAX
☆30Updated this week
Alternatives and similar repositories for unxt:
Users that are interested in unxt are comparing it to the libraries listed below
- Curated list of astronomy/astrophysics code packages using JAX☆37Updated 2 months ago
- Coordinates in JAX☆16Updated 2 weeks ago
- Astronomy Research + JAX Meeting 2024☆13Updated 3 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆34Updated 2 weeks ago
- Numerical quadrature with JAX☆53Updated this week
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆14Updated 3 months ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆25Updated 3 months ago
- A fast, differentiable, and extensible public BBN code☆11Updated last week
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆36Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated 11 months ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆36Updated 8 months ago
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆13Updated 11 months ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆60Updated this week
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆24Updated 2 weeks ago
- Differentiable Likelihood for CMB Analysis☆15Updated last week
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆29Updated 2 weeks ago
- Robust ML in Astro☆51Updated this week
- Rapid population synthesis code for binary black hole mergers in dynamical environments.☆12Updated 9 months ago
- An experiment: emcee implemented in JAX☆25Updated 2 years ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆14Updated 2 months ago
- Differentiable Cosmological Forward Model☆78Updated 2 months ago
- jaxspec is an X-ray spectra Bayesian analysis package, relying on JAX to enable just in time compilation☆29Updated this week
- Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation☆23Updated last week
- Differentiable Stellar Population Synthesis☆26Updated last week
- A general-purpose algorithm for finding astrophysically-relevant clusters from point-cloud data.☆10Updated last week
- nessai: Nested Sampling with Artificial Intelligence☆38Updated 2 weeks ago
- Coverage tests to check the quality of your posterior estimators.☆30Updated 6 months ago