Skip to content

Add high level autograd functions #30632

@albanD

Description

@albanD

It would be very useful to have high level autograd functions such as:

  • jacobian
  • hessian
  • jvp
  • vjp

The questions are:

  • Where should they go? torch.autograd.??
  • What kind of api we want to provide: should they be function to function mapping? Or should they be function + value to value mappings?

cc @ezyang @ssnl @albanD @zou3519 @gqchen

Metadata

Metadata

Assignees

Labels

enhancementNot as big of a feature, but technically not a bug. Should be easy to fixmodule: autogradRelated to torch.autograd, and the autograd engine in generaltriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions