Dynamax: A Python package for probabilistic state space modeling with JAX. - Scott W. Linderman; Peter G. Chang; Giles Harper-Donnelly; Aleyna Kara; Xinglong Li; Gerardo Duran-Martin; Kevin Murphy 002 (J. Open Source Softw., 2025)