sstrax: Fast stellar stream modelling in JAX
Abstract
sstrax provides fast simulations of Milky Way stellar stream formation. Using JAX (ascl:2111.002) acceleration to support code compilation, sstrax forward models all aspects of stream formation, including evolution in gravitational potentials, tidal disruption and observational models, in a fully modular way. Although sstrax is a standalone python package, it was also developed to integrate directly with the Albatross (ascl:2306.009) inference pipeline, which performs inference on all relevant aspects of the stream model.
- Publication:
-
Astrophysics Source Code Library
- Pub Date:
- June 2023
- Bibcode:
- 2023ascl.soft06008A
- Keywords:
-
- Software