Fix the norm of complex numbers in clipping#161

wdphy16 wants to merge 3 commits intogoogle-deepmind:mastergoogle-deepmind/optax:masterfrom

wdphy16:complex_normCopy head branch name to clipboard