SBIAX: Density-estimation simulation-based inference in JAX. - Jed Homer; Oliver Friedrich (J. Open Source Softw., 2025)