normalize

normalize(unconstrained, scales)

Divide unconstrained params by their scales → O(1) optimizer space.

Parameters

Name Type Description Default
unconstrained dict[str, Array] in unconstrained space. required
scales dict[str, Array] from :func:compute_param_scales. required

Returns

Name Type Description
dict[str, Array] with each value divided by its scale.