We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
cf71bab
Moved eps outside sqrt so that this implementation behaves similarly to reference implementation (see google-deepmind/optax#401)
