validate

pmrf.modules.validate(tree: PyTree)

Validate a parameter PyTree.

Any PyTree structure is accepted. ParamRF modules nested anywhere within it additionally retain their stricter field validation, which prevents raw JAX arrays from ambiguously representing either fixed or free parameters.