metaopt/torchopt

[Feature Request] Implicit gradient support

Benjamin-eecs opened this issue · 0 comments

Motivation

for imaml and model-based meta-rl algos

Solution

  • linear system solver for hessian vector product
  • jaxopt decorator
  • imaml example
  • implicit differentiation tutorial
  • api docs
  • readme

Checklist

  • I have checked that there is no similar issue in the repo (required)