warp
warp copied to clipboard
[REQ] Add option to replay forward dynamic loop in the backward pass
Description
See https://nvidia.github.io/warp/modules/differentiability.html#dynamic-loops
The last example requires the user to migrate the dynamic loop to a separate @wp.func