unxt

unxt is a Python package for unit-aware computing with JAX, which is a high-performance numerical computing library that enables automatic differentiation and just-in-time compilation to accelerate code execution on multiple compute architectures. 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.