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