freeze (pmrf.freeze)

pmrf.freeze(model: Any)

Freezes a model (or any JAX PyTree) and returns the frozen model.

This can be used to freeze models to make them non-optimizable, but should also be used as a field converter (using prf.field(converter=prf.freeze)) when storing raw arrays within in a model.