Implement ANO in JAX!

[tinker495]

optax PR link

Hi! I found the paper interesting and added the optimizer to optax, the official optimizer framework for JAX.
I'm opening this issue to ask for confirmation from the author of the paper to make the PR official. Please check it out and see if you can find any deviations in the implementation!

[Adrienkgz]

Hi @tinker495, thank you for your contribution and for taking the time to implement this optimizer. I really appreciate your effort in bringing this optimizer to Optax!