allen-adastra / xarray_jax
Simple Xarray + JAX Integration
☆13Updated last week
Related projects ⓘ
Alternatives and complementary repositories for xarray_jax
- Numerical quadrature with JAX☆43Updated 2 weeks ago
- Interpolation and function approximation with JAX☆132Updated 2 weeks ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated last month
- JAX + Units☆43Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Manipulating ragged arrays in an Array API compliant way.☆30Updated last month
- Python wrapper to some of Quadpack (numerical integration), which can be called from within numba functions.☆25Updated 6 months ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆11Updated 8 months ago
- Probabilistic Programming and Nested sampling in JAX☆147Updated this week
- Unitful Quantities in JAX☆14Updated this week
- My CV☆36Updated this week
- Differentiable and accelerated spherical transforms with JAX☆134Updated 2 weeks ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆18Updated 3 weeks ago
- Unleash the true power of scheduling☆24Updated 3 weeks ago
- Fully and Partially Bayesian Neural Nets☆22Updated this week
- Stencil computations in JAX☆66Updated last year
- PyMC Extras extracted from the "exoplanet" library☆19Updated last month
- Learning materials for scientific software development☆78Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- Taylor Swift-inspired color maps☆81Updated 3 months ago
- Minterpy - Multivariate interpolation in Python.☆33Updated last month
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆21Updated last week
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆56Updated this week
- A dead simple Python package for creating custom JAX pytree objects☆61Updated 4 months ago
- differentiable (binned) likelihoods with JAX☆18Updated 2 weeks ago
- An experiment: emcee implemented in JAX☆25Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- A differentiable cosmology library in JAX☆176Updated 3 months ago
- Stainless neural networks in JAX☆30Updated 3 weeks ago
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week