jinnax

Python package jinnax provides functions to generate synthetic data and easily train Physics-informed Neural Networks in JAX. It has the modules data and nn for data generation and neural network training.

A tutorial of jinnax is available at Google Colab where more details about its data generating and training functions can be found.

View Source

1"""
2 Python package `jinnax` provides functions to generate synthetic data and easily train
3 Physics-informed Neural Networks in `JAX`. It has the modules `data` and `nn` for data generation
4 and neural network training.
5
6 A tutorial of `jinnax` is available at [Google Colab](https://colab.research.google.com/github/dmarcondes/JINNAX/blob/master/Tutorial_PINN_simulation_with_JINNAX.ipynb)
7 where more details about its data generating and training functions can be found.
8 """