types.equinox_parameter.EquinoxParameter
types.equinox_parameter.EquinoxParameter()Mark an Equinox module’s inexact array leaves as trainable.
Space does not support axes inside an EquinoxParameter. PyTree-visible axes are rejected explicitly; fields declared with eqx.field(static=True) are preserved as static module metadata and are not inspected or swept. Place all Space axes outside the module.