unxt: A Python package for unit-aware computing with JAX
unxt is a Python package for unit-aware computing with JAX. unxt is built on top of quax, which provides a framework for building array-like objects that can be used with JAX. unxt extends quax to provide support for unit-aware computing using the astropy.units package as a units backend. unxt provides seamless integration of physical units into high performance numerical computations, significantly enhancing the capabilities of JAX for scientific applications.
Authors
- Jacob Nibauer (ORCID: https://orcid.org/0000-0001-8042-5794)
- Adrian M. Price-Whelan (ORCID: https://orcid.org/0000-0003-0872-7098)
- Nathaniel Starkman (ORCID: https://orcid.org/0000-0003-3954-3291)
Institutions
- Princeton University (US)
- Kavli Institute for Particle Astrophysics and Cosmology (US)
- Flatiron Health (United States) (US)
- Flatiron Institute (US)
- Massachusetts Institute of Technology (US)
Publication Details
- Journal
- The Journal of Open Source Software
- Published
- 2025-03-28
- DOI
- https://doi.org/10.21105/joss.07771
- Primary Topic
- Computational Physics and Python Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Brinson Foundation
- Flatiron Health