English

unxt: A Python package for unit-aware computing with JAX

Instrumentation and Methods for Astrophysics 2026-03-11 v1 Astrophysics of Galaxies Computational Physics

Abstract

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.

Cite

@article{arxiv.2603.08770,
  title  = {unxt: A Python package for unit-aware computing with JAX},
  author = {Nathaniel Starkman and Adrian Price-Whelan and Jake Nibauer},
  journal= {arXiv preprint arXiv:2603.08770},
  year   = {2026}
}

Comments

4 pages, software

R2 v1 2026-07-01T11:10:55.580Z