A set of tools for developing new methods and techniques in physics informed neural networks written in jax. - View it on GitHub