Open
Description
jaxopt is being merged with optax. All the features we use are in optax anyway, so reimplement using the optax-native functions instead.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Metadata
Assignees
Labels
No labels