Annotated
parax.annotated.AbstractAnnotated
Bases: Module, Generic[T]
An abstract interface for an Equinox module that is annotated with arbitrary metadata.
In Parax, metadata is typically used to store units, descriptions, tags, or optimization hints alongside the underlying JAX array.
Attributes:
| Name | Type | Description |
|---|---|---|
metadata |
AbstractVar[T]
|
Returns the underlying metadata. |