google / google/jaxopt

Add Broyden in the DEQ example

Open
#426 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.1k
Forks
76
Avg merge
2d 21h
Merged PRs (30d)
1

Description

Once [the Broyden PR](https://github.com/google/jaxopt/pull/416) is merged, I would love to contribute a DEQ example maybe inspired by https://github.com/akbir/deq-jax but probably much simpler to keep it in a single notebook.
I think a very simple thing on MNIST would do the trick without necessarily using complex multiscale DEQs.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.