Efficient PINNs and Natural Gradients in JAX

Talk presenting an efficient JAX implementation of natural gradient methods for Physics-Informed Neural Networks (PINNs).

Slides