mapparam: Mapped parameters and parameter maps#
Parameter mappers / mapped parameters for equinox modules.
Similar to PyTorch’s torch.nn.utils.parametrize.
MappedParameterDomainMappedParameterIdentityMappedParameterAffineMappedParameterTanhMappedParameterAmplitudeTanhMappedParameterMappedLogitsNormSphereParameterProbabilitySimplexParameterAmplitudeProbabilitySimplexParameterOrthogonalParameterIsochoricParameterClipRenormaliseForcePositiveDefinitePhaseAmplitudeMixin