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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Starkman, Nathaniel, Price-Whelan, Adrian, Nibauer, Jake
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912957393272832
author Starkman, Nathaniel
Price-Whelan, Adrian
Nibauer, Jake
author_facet Starkman, Nathaniel
Price-Whelan, Adrian
Nibauer, Jake
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2603_08770
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle unxt: A Python package for unit-aware computing with JAX
Starkman, Nathaniel
Price-Whelan, Adrian
Nibauer, Jake
Instrumentation and Methods for Astrophysics
Astrophysics of Galaxies
Computational Physics
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.
title unxt: A Python package for unit-aware computing with JAX
topic Instrumentation and Methods for Astrophysics
Astrophysics of Galaxies
Computational Physics
url https://arxiv.org/abs/2603.08770