Qazalbash / jaxamplerLinks
An open-source JAX-based statistical sampling toolkit ๐งช
โ10Updated last year
Alternatives and similar repositories for jaxampler
Users that are interested in jaxampler are comparing it to the libraries listed below
Sorting:
- An open-source JAX-based gravitational-wave population inference toolkit for parametric modelsโ23Updated this week
- Torch utilities for doing machine learning in gravitational wave physicsโ32Updated last week
- Code repository for CIERA's Research Experiences in Astronomy at CIERA for High School Students (REACH) Programโ14Updated 5 months ago
- Performing parameter estimation on gravitational wave data with machine learningโ22Updated last week
- JAX-powered Cosmological Particle-Mesh N-body Solverโ59Updated last week
- JAX bindings for the NVIDIA cuDecomp libraryโ50Updated 2 months ago
- GWInferno: Gravitational-Wave Hierarchical Inference with NumPyroโ21Updated this week
- Differentiable Gravitational Waveforms with JAXโ69Updated 5 months ago
- Curated list of astronomy/astrophysics code packages using JAXโ51Updated 2 weeks ago
- Gravitational-wave data analysis tools in Jaxโ79Updated 4 months ago
- โ10Updated last year
- CPU/GPU agnostic gravitational-wave population inferenceโ56Updated last week
- Package for data-driven and phenomenological gravitational waveform modelsโ12Updated 10 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilationโ75Updated 2 months ago
- Detecting binary black hole mergers in LIGO with neural networksโ17Updated 2 months ago
- nessai: Nested Sampling with Artificial Intelligenceโ43Updated last week
- Material for the CMU Quarks2Cosmos Conference Data Challengesโ14Updated 2 years ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selectionโ73Updated last week
- Differentiable and accelerated wavelet transform on the sphere with JAXโ24Updated 7 months ago
- Coverage tests to check the quality of your posterior estimators.โ32Updated last year
- Dingo: Deep inference for gravitational-wave observationsโ106Updated last week
- MCHMC: sampler from an arbitrary differentiable distributionโ75Updated 2 weeks ago
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyroโ16Updated 10 months ago
- Generative models with JAX.โ20Updated 11 months ago
- A Jax wrapper for cudaKDTreeโ11Updated 3 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) libraryโ108Updated last week
- Unitful Quantities in JAXโ49Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.โ20Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.โ138Updated last month
- Fast, lightweight and parallelised simulation-based inference in JAX.โ22Updated 2 months ago