Use new parametrization based weight norm if available (#24030)
* Use new parametrization based weight norm if available
See https://github.com/pytorch/pytorch/pull/103001
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
* handle copies
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
* black
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
---------
Signed-off-by: Edward Z. Yang <ezyang@meta.com>