optax and jaxopt

These two libraries are **complements**, as Optax focuses on gradient-based optimization algorithms and utilities, while JAXopt provides a broader range of differentiable and hardware-accelerated optimizers, including non-gradient-based methods, which can leverage or be combined with Optax's gradient processing capabilities.