GalacticDynamics / unxt
Unitful Quantities in JAX
☆32Updated last 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☆38Updated 3 months ago
- Astronomy Research + JAX Meeting 2024☆13Updated 5 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆36Updated 3 weeks ago
- Coordinates in JAX☆17Updated 2 weeks ago
- Numerical quadrature with JAX☆53Updated this week
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆14Updated 3 weeks ago
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆38Updated last week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆60Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated last year
- nessai: Nested Sampling with Artificial Intelligence☆38Updated 2 weeks ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆37Updated 2 weeks ago
- Differentiable Cosmological Forward Model☆78Updated 3 weeks ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆24Updated last month
- JAX-based linear Einstein-Boltzmann solver for cosmology☆15Updated this week
- A fast, differentiable, and extensible public BBN code☆11Updated 3 weeks ago
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆13Updated last year
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 11 months ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆26Updated 4 months ago
- Neural Network-Boosted Importance Nested Sampling for Bayesian Statistics☆83Updated last month
- JAX bindings for the NVIDIA cuDecomp library☆32Updated 3 weeks ago
- ☆16Updated 4 years ago
- Gradient Based Nested Sampling☆20Updated last year
- A general-purpose algorithm for finding astrophysically-relevant clusters from point-cloud data.☆11Updated 2 weeks ago
- Differentiable Gravitational Waveforms with JAX☆55Updated 7 months ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆29Updated 2 weeks ago
- Differentiable Likelihood for CMB Analysis☆17Updated last month
- An experiment: emcee implemented in JAX☆25Updated 2 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated 2 weeks ago