jaxoplanet.units.field

Functions

field(→ Any)

A custom Equinox field with support for units

Module Contents

jaxoplanet.units.field.field(*, units: Any = None, strict: bool = False, **kwargs: Any) → Any

A custom Equinox field with support for units

This is a wrapper around equinox.field() that adds support for units. The units argument is used to specify the units of the field either as a plain Unit or string, or as a Pytree with the same structure as the expected input.