Replace JAXopt with Optax

Open ankit27kh opened this issue 2 years ago • 1 comments

JAXopt is going to be merged into Optax. https://github.com/google/jaxopt?tab=readme-ov-file#%EF%B8%8F-we-are-in-the-process-of-merging-jaxopt-into-optax-because-of-this-jaxopt-is-now-in-maintenance-mode-and-we-will-not-be-implementing-new-features-%EF%B8%8F

Feb 08 '24 07:02 ankit27kh

Thank you for raising this issue!

Feb 08 '24 15:02 dime10

This was completed in #808 :)

Jun 24 '24 17:06 dime10