Skip to article frontmatterSkip to article content
Site not loading correctly?

This may be due to an incorrect BASE_URL configuration. See the MyST Documentation for reference.

Dimensional quantities

Owner import path

jaxstro.quantity

Purpose

Implemented dimensional quantity evaluation, unit parsing and registries, conversion, serialization, constants, and equivalencies.

Public records and callables

Quantity, Unit, Dimension, UnitRegistry, their typed errors, concrete units and dimensions, get_unit, format_unit, parse_unit, serialization helpers, and the public constants, equivalencies, astro, bases, and units modules.

Shape and dtype expectations

Quantity values are JAX-compatible arrays. Unit and dimension objects are static metadata at traced boundaries.

JAX transforms and AD classification

Value conversion composes with JAX transforms when units are fixed. Parsing, registry mutation, and dimensional validation are host-side.

Failure behavior

Dimension, conversion, equivalency, parsing, and registry failures use explicit typed exceptions. Ecosystem adoption remains deferred; this page is not a cutover announcement.

See Unit systems and quantity adoption, Quantities, units, and dimensional boundaries, and ADR 0006 - Build own Quantity, not unxt/quax.

Canonical import example

from jaxstro.quantity import Quantity