Jim 🚬¤
A JAX-based gravitational-wave inference toolkit¤
Jim is a JAX-based toolkit for Bayesian parameter estimation of gravitational-wave sources. It pairs differentiable waveform models from ripple with GPU-accelerated JAX-based samplers, enabling massively parallel inference.
Supported samplers:
- flowMC — normalizing-flow-enhanced MCMC with optional parallel tempering.
- BlackJAX NS-AW — nested sampling described in Prathaban et al. 2025 (arXiv:2509.04336).
- BlackJAX NSS — nested slice sampling.
- BlackJAX SMC — sequential Monte Carlo with optional adaptive tempering and persistent sampling.
Warning
Jim has not yet reached v1.0.0 and the API may change. Use at your own risk. Consider pinning to a specific version if you need API stability.
Documentation¤
- Installation — How to install Jim
- Quick Start — Run your first analysis with
jim-run - Tutorials — Hands-on worked examples
- Guides — In-depth coverage of each module and the CLI config reference
- FAQ — Answers to common questions
- Citing Jim — How to cite Jim
- Contributing to Jim — How to contribute to Jim