optax-adan (pypi): An implementation of adan optimization algorithm for optax.
Keywords: deeplearning|jax|optax|optimization-algorithms|optimization-methods