Is it possible to do Nested AD ~elegantly~ in Julia? (PINNs)

Lux is still a solution, and it’s still the case that nested AD is a terrible way to do PINNs anyways.

2 Likes