Expanded implementation of the Neural Tangents library for neural network solvers of differential equations.

Use stax.Tailor and stax.Deriv1d provided in this expansion to consturct the desired NTK.

Classes to reproduce the experimental results are included in example.py