mapparam
: Mapped parameters and parameter maps#
Parameter mappers / mapped parameters for equinox
modules.
Similar to PyTorch’s torch.nn.utils.parametrize
.
MappedParameter
DomainMappedParameter
IdentityMappedParameter
AffineMappedParameter
TanhMappedParameter
AmplitudeTanhMappedParameter
MappedLogits
NormSphereParameter
ProbabilitySimplexParameter
AmplitudeProbabilitySimplexParameter
OrthogonalParameter
IsochoricParameter
Clip
Renormalise
ForcePositiveDefinite
PhaseAmplitudeMixin