Optax is a gradient processing and optimization library for JAX.
By chatting or signing in you agree to the Terms and chat-message logging (revocable in History).