Hello,
Thanks a lot!

You would need a function that computes this second derivative

How does it work exactly ? Should I write a backward for the backward ? I didn’t find many implementations of such layers.

the first derivative to be implemented in a differentiable way with the autograd engine.

Do you mean that as long as the backward is implemented in a differentiable way, it will automatically handle double diff ?
thank you