subclass_zoo icon indicating copy to clipboard operation
subclass_zoo copied to clipboard

Published 20 hours ago • verified publisher icon albanD

custom_vjp bug

Open zou3519 opened this issue 2 years ago • 0 comments

There's a bug in simple functorch's custom_vjp. In particular, we would like to learn what it would take to get simple functorch's custom_vjp to work like JAX's custom_vjp w.r.t. to the behavior towards intermediate Tensors.

TODO: we should try to fix the bugs mentioned.

zou3519 avatar May 02 '22 21:05 zou3519