LICENSE
README.md
pyproject.toml
setup.py
pinnsjax/__init__.py
pinnsjax/train.py
pinnsjax.egg-info/PKG-INFO
pinnsjax.egg-info/SOURCES.txt
pinnsjax.egg-info/dependency_links.txt
pinnsjax.egg-info/entry_points.txt
pinnsjax.egg-info/requires.txt
pinnsjax.egg-info/top_level.txt
pinnsjax/conf/__init__.py
pinnsjax/conf/data/__init__.py
pinnsjax/conf/extras/__init__.py
pinnsjax/conf/hparams_search/__init__.py
pinnsjax/conf/hydra/__init__.py
pinnsjax/conf/mesh/__init__.py
pinnsjax/conf/model/__init__.py
pinnsjax/conf/net/__init__.py
pinnsjax/conf/paths/__init__.py
pinnsjax/conf/spatial_domain/__init__.py
pinnsjax/conf/time_domain/__init__.py
pinnsjax/conf/trainer/__init__.py
pinnsjax/data/__init__.py
pinnsjax/data/pinn_datamodule.py
pinnsjax/data/domains/__init__.py
pinnsjax/data/domains/point_cloud.py
pinnsjax/data/domains/spatial.py
pinnsjax/data/domains/time.py
pinnsjax/data/mesh/__init__.py
pinnsjax/data/mesh/mesh.py
pinnsjax/data/sampler/__init__.py
pinnsjax/data/sampler/boundary_condition.py
pinnsjax/data/sampler/initial_condition.py
pinnsjax/data/sampler/mesh_sampler.py
pinnsjax/data/sampler/sampler_base.py
pinnsjax/models/__init__.py
pinnsjax/models/pinn_module.py
pinnsjax/models/net/__init__.py
pinnsjax/models/net/neural_net.py
pinnsjax/models/runge_kutta/__init__.py
pinnsjax/models/runge_kutta/runge_kutta.py
pinnsjax/trainer/__init__.py
pinnsjax/trainer/trainer.py
pinnsjax/utils/__init__.py
pinnsjax/utils/gradient.py
pinnsjax/utils/module_fn.py
pinnsjax/utils/plotting.py
pinnsjax/utils/pylogger.py
pinnsjax/utils/rich_utils.py
pinnsjax/utils/utils.py