Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jacobian in Zygote with Params #413

Closed
theogf opened this issue Dec 5, 2019 · 2 comments · Fixed by #890
Closed

Jacobian in Zygote with Params #413

theogf opened this issue Dec 5, 2019 · 2 comments · Fixed by #890

Comments

@theogf
Copy link

theogf commented Dec 5, 2019

I simply love the Flux+Zygote flexibility for extracting parameters from object and get the appropriate gradients directly in the same structure (Grads).
However it would be nice to have the same feature for jacobians with Params. I saw the already open issues #98 and #51. But the proposed workaround is only valid for arrays.

@MikeInnes
Copy link
Member

It'd make some sense to do jacobians this way. You could build it on top of gradients pretty easily, I think.

@mcabbott
Copy link
Member

mcabbott commented Dec 5, 2019

I doodled something together a little while ago, link, maybe I should make a PR...

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
3 participants