jaxrts: A Python package for simulating X-ray Thomson scattering spectra from dense plasmas using jax
Description
Over the last decades, X-ray Thomson scattering has become a valuable experimental technique to study dense plasma and warm dense matter states. Designing experiments and interpreting data typically requires the simulation of spectra for given sets of plasma parameters. Many models applicable to different conditions have been formulated by the community. However, an open, broadly available code combining these efforts has been lacking. In this work, we introduce the open-source simulation package jaxrts, a Python implementation incorporating several models, designed to be modular, extensible, transparent, and user-friendly. The software unites Python's flexibility with computational efficiency through jax's just-in-time compilation and GPU acceleration.