prism
prism copied to clipboard
Supporting orthogonal parameterization
https://pytorch.org/docs/stable/generated/torch.nn.utils.parametrizations.orthogonal.html
geotorch might be useful https://github.com/Lezcano/geotorch