LICENSE
README.md
pyproject.toml
surkit/backend/__init__.py
surkit/backend/backend.py
surkit/backend/jax_bkd.py
surkit/backend/oneflow_bkd.py
surkit/backend/pytorch_bkd.py
surkit/conditions/__init__.py
surkit/conditions/hard_condition.py
surkit/conditions/icbc.py
surkit/conditions/icbc2d.py
surkit/data/__init__.py
surkit/data/dataset.py
surkit/data/processing.py
surkit/data/sampling.py
surkit/gradient/__init__.py
surkit/gradient/sobelfilter.py
surkit/losses/__init__.py
surkit/losses/losses_jax.py
surkit/losses/losses_oneflow.py
surkit/losses/losses_pytorch.py
surkit/nn/__init__.py
surkit/nn/jax/__init__.py
surkit/nn/jax/activations.py
surkit/nn/jax/bayes_cnn.py
surkit/nn/jax/bayes_dense_ed.py
surkit/nn/jax/bayes_layer.py
surkit/nn/jax/bayes_nn.py
surkit/nn/jax/cnn.py
surkit/nn/jax/dense_ed.py
surkit/nn/jax/fnn.py
surkit/nn/jax/initializers.py
surkit/nn/jax/nn_ensemble.py
surkit/nn/jax/pooling.py
surkit/nn/jax/transform.py
surkit/nn/jax/utils.py
surkit/nn/oneflow/__init__.py
surkit/nn/oneflow/activations.py
surkit/nn/oneflow/bayes_cnn.py
surkit/nn/oneflow/bayes_dense_ed.py
surkit/nn/oneflow/bayes_layer.py
surkit/nn/oneflow/bayes_nn.py
surkit/nn/oneflow/cnn.py
surkit/nn/oneflow/dense_ed.py
surkit/nn/oneflow/fnn.py
surkit/nn/oneflow/initializers.py
surkit/nn/oneflow/nn_ensemble.py
surkit/nn/oneflow/pooling.py
surkit/nn/oneflow/transform.py
surkit/nn/oneflow/utils.py
surkit/nn/pytorch/__init__.py
surkit/nn/pytorch/activations.py
surkit/nn/pytorch/bayes_cnn.py
surkit/nn/pytorch/bayes_dense_ed.py
surkit/nn/pytorch/bayes_layer.py
surkit/nn/pytorch/bayes_nn.py
surkit/nn/pytorch/cnn.py
surkit/nn/pytorch/dense_ed.py
surkit/nn/pytorch/fnn.py
surkit/nn/pytorch/initializers.py
surkit/nn/pytorch/nn_ensemble.py
surkit/nn/pytorch/pooling.py
surkit/nn/pytorch/transform.py
surkit/optimizer/__init__.py
surkit/optimizer/optimizer_jax.py
surkit/optimizer/optimizer_oneflow.py
surkit/optimizer/optimizer_pytorch.py
surkit/pde/__init__.py
surkit/pde/pde.py
surkit/surkit.egg-info/PKG-INFO
surkit/surkit.egg-info/SOURCES.txt
surkit/surkit.egg-info/dependency_links.txt
surkit/surkit.egg-info/requires.txt
surkit/surkit.egg-info/top_level.txt
surkit/train/__init__.py
surkit/train/train_jax.py
surkit/train/train_oneflow.py
surkit/train/train_pytorch.py
surkit/utils/__init__.py
surkit/utils/math_utils.py
surkit/utils/parse.py