Skip to content

Functorch update (PyTorch 2)#41

Draft
rfeinman wants to merge 9 commits into
mainfrom
functorch
Draft

Functorch update (PyTorch 2)#41
rfeinman wants to merge 9 commits into
mainfrom
functorch

Conversation

@rfeinman

Copy link
Copy Markdown
Owner

Incorporate new PyTorch 2 torch.func module for all jacobian & hessian computations. Look for opportunities to use forward-mode autograd (vs reverse-mode) for efficiency.

@rfeinman

Copy link
Copy Markdown
Owner Author

TODO: update lstsq linear operator code.

@rfeinman rfeinman marked this pull request as draft January 15, 2026 21:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant