ott icon indicating copy to clipboard operation
ott copied to clipboard
verified publisher icon ott-jax

JaxOpt is no longer maintained

Open Zethson opened this issue 10 months ago • 1 comments 

../../miniforge3/envs/pertpy/lib/python3.12/site-packages/jaxopt/__init__.py:59
  python3.12/site-packages/jaxopt/__init__.py:59: DeprecationWarning: JAXopt is no longer maintained. See https://docs.jax.dev/en/latest/ for alternatives.
    warnings.warn(

JaxOpt is a dependency of ott-jax and should ideally be replaced with something else.

Zethson avatar Jun 15 '25 14:06 Zethson

this is a good point. We're mostly using Jaxopt to run proximal gradient descent and some h-transforms. Maybe we should copy-paste their code as they haven't moved what we're interested in in optax.

marcocuturi avatar Jun 15 '25 19:06 marcocuturi