maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Published 20 hours ago • verified publisher icon google

Change l2norm to use jnp.sqrt

Open raymondzouu opened this issue 10 months ago • 0 comments

Changing l2norm to use jnp.sqrt instead of **0.5. Seeing a speed up on small examples: https://screenshot.googleplex.com/A3GjjWQq5Dhes9b Colab notebook: http://shortn/_p369zYcGI2

raymondzouu avatar Apr 24 '24 06:04 raymondzouu