.envrc
.gitignore
.pre-commit-config.yaml
CHANGELOG.md
LICENSE
README.md
environment.yaml
init_conda_venv.sh
init_python_venv.sh
noxfile.py
pyproject.toml
.github/workflows/dev-ci-actions.yaml
.github/workflows/docs-build-deploy.yaml
.github/workflows/latest-tests.yaml
.github/workflows/main-ci-actions.yaml
.github/workflows/pinned-tests.yaml
.github/workflows/publish-package.yaml
.github/workflows/quality-typing-checks.yaml
.github/workflows/release-ci-actions.yaml
docs/Makefile
docs/make.bat
docs/source/conf.py
docs/source/index.md
docs/source/_static/PhysicsML_Logo-01.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo full dark-cropped.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo full dark.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo full light-cropped.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo full light.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo text dark.svg
docs/source/_static/PhysicsML_Logo_PhyML_logo text light.svg
docs/source/pages/datasets/qm_datasets.md
docs/source/pages/features/gdb9_trunc.parquet
docs/source/pages/features/how_to_add_reps.md
docs/source/pages/features/intro.md
docs/source/pages/models/allegro.md
docs/source/pages/models/ani.md
docs/source/pages/models/egnn.md
docs/source/pages/models/how_to_add_models.md
docs/source/pages/models/intro.md
docs/source/pages/models/mace.md
docs/source/pages/models/nequip.md
docs/source/pages/models/tensor_net.md
docs/source/pages/philosophy/molflux.md
docs/source/pages/philosophy/philosophy.md
docs/source/pages/plugins/.gitignore
docs/source/pages/plugins/alanine-dipeptide-truncated.pdb
docs/source/pages/plugins/ase.md
docs/source/pages/plugins/openmm.md
docs/source/pages/structure/lightning_layer.md
docs/source/pages/structure/molflux_layer.md
docs/source/pages/structure/physicsml_structure.md
docs/source/pages/structure/torch_layer.md
docs/source/pages/structure/transfer_learning.md
docs/source/pages/tutorials/.gitignore
docs/source/pages/tutorials/ani1x_energy_forces_training.md
docs/source/pages/tutorials/ani1x_truncated.parquet
docs/source/pages/tutorials/gdb9_training.md
docs/source/pages/tutorials/gdb9_trunc.parquet
docs/source/pages/tutorials/gdb9_uncertainty.md
docs/source/pages/tutorials/transfer_learning.md
pinned-versions/3.10/lockfile.ase.txt
pinned-versions/3.10/lockfile.core.txt
pinned-versions/3.10/lockfile.openeye.txt
pinned-versions/3.10/lockfile.openmm.txt
pinned-versions/3.10/lockfile.rdkit.txt
pinned-versions/3.11/lockfile.ase.txt
pinned-versions/3.11/lockfile.core.txt
pinned-versions/3.11/lockfile.openeye.txt
pinned-versions/3.11/lockfile.openmm.txt
pinned-versions/3.11/lockfile.rdkit.txt
src/physicsml/__init__.py
src/physicsml/py.typed
src/physicsml/utils.py
src/physicsml/version.py
src/physicsml.egg-info/PKG-INFO
src/physicsml.egg-info/SOURCES.txt
src/physicsml.egg-info/dependency_links.txt
src/physicsml.egg-info/entry_points.txt
src/physicsml.egg-info/requires.txt
src/physicsml.egg-info/top_level.txt
src/physicsml/backends/__init__.py
src/physicsml/backends/backend_selector.py
src/physicsml/backends/openeye_backend.py
src/physicsml/backends/rdkit_backend.py
src/physicsml/featurisation/__init__.py
src/physicsml/featurisation/physicsml_features/atom_features.py
src/physicsml/featurisation/physicsml_features/bond_features.py
src/physicsml/featurisation/physicsml_features/physicsml_features.py
src/physicsml/lightning/__init__.py
src/physicsml/lightning/config.py
src/physicsml/lightning/datamodule.py
src/physicsml/lightning/model.py
src/physicsml/lightning/model_uncertainty.py
src/physicsml/lightning/module.py
src/physicsml/lightning/pre_batching_in_memory.py
src/physicsml/lightning/pre_batching_on_disk.py
src/physicsml/lightning/graph_datasets/__init__.py
src/physicsml/lightning/graph_datasets/graph_dataset.py
src/physicsml/lightning/graph_datasets/neighbourhood_list_torch.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/__init__.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/geometry.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/linked_cell.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/naive_impl.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/neighbor_list.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/timer.py
src/physicsml/lightning/graph_datasets/torch_nl_vendored/utils.py
src/physicsml/lightning/losses/__init__.py
src/physicsml/lightning/losses/construct_loss.py
src/physicsml/lightning/losses/loss_base.py
src/physicsml/lightning/losses/masked_mse_loss.py
src/physicsml/lightning/losses/multitask_losses.py
src/physicsml/lightning/losses/serial_bce_w_logits_loss.py
src/physicsml/lightning/losses/stock_losses.py
src/physicsml/lightning/losses/weighted_mse_loss.py
src/physicsml/models/__init__.py
src/physicsml/models/prism.py
src/physicsml/models/utils.py
src/physicsml/models/allegro/__init__.py
src/physicsml/models/allegro/allegro_prism.py
src/physicsml/models/allegro/mean_var/__init__.py
src/physicsml/models/allegro/mean_var/default_configs.py
src/physicsml/models/allegro/mean_var/mean_var_allegro_model.py
src/physicsml/models/allegro/mean_var/mean_var_allegro_module.py
src/physicsml/models/allegro/modules/__init__.py
src/physicsml/models/allegro/modules/allegro.py
src/physicsml/models/allegro/modules/channels.py
src/physicsml/models/allegro/modules/contract.py
src/physicsml/models/allegro/modules/cutoffs.py
src/physicsml/models/allegro/modules/fc.py
src/physicsml/models/allegro/modules/layout.py
src/physicsml/models/allegro/modules/linear.py
src/physicsml/models/allegro/modules/radial.py
src/physicsml/models/allegro/modules/spmm.py
src/physicsml/models/allegro/supervised/__init__.py
src/physicsml/models/allegro/supervised/allegro_model.py
src/physicsml/models/allegro/supervised/allegro_module.py
src/physicsml/models/allegro/supervised/default_configs.py
src/physicsml/models/ani/__init__.py
src/physicsml/models/ani/ani_1_2_defaults.py
src/physicsml/models/ani/ani_datamodule.py
src/physicsml/models/ani/ani_dataset.py
src/physicsml/models/ani/ani_prism.py
src/physicsml/models/ani/config.py
src/physicsml/models/ani/ensemble/default_configs.py
src/physicsml/models/ani/ensemble/ensemble_ani_model.py
src/physicsml/models/ani/ensemble/ensemble_ani_module.py
src/physicsml/models/ani/mean_var/__init__.py
src/physicsml/models/ani/mean_var/default_configs.py
src/physicsml/models/ani/mean_var/mean_var_ani_model.py
src/physicsml/models/ani/mean_var/mean_var_ani_module.py
src/physicsml/models/ani/modules/__init__.py
src/physicsml/models/ani/modules/aev.py
src/physicsml/models/ani/supervised/__init__.py
src/physicsml/models/ani/supervised/ani_model.py
src/physicsml/models/ani/supervised/ani_module.py
src/physicsml/models/ani/supervised/default_configs.py
src/physicsml/models/egnn/__init__.py
src/physicsml/models/egnn/egnn_prism.py
src/physicsml/models/egnn/egnn_utils.py
src/physicsml/models/egnn/adapter/__init__.py
src/physicsml/models/egnn/adapter/adapter_egnn_model.py
src/physicsml/models/egnn/adapter/adapter_egnn_module.py
src/physicsml/models/egnn/adapter/adapter_egnn_utils.py
src/physicsml/models/egnn/adapter/default_configs.py
src/physicsml/models/egnn/mean_var/__init__.py
src/physicsml/models/egnn/mean_var/default_configs.py
src/physicsml/models/egnn/mean_var/mean_var_egnn_model.py
src/physicsml/models/egnn/mean_var/mean_var_egnn_module.py
src/physicsml/models/egnn/multi_graph/__init__.py
src/physicsml/models/egnn/multi_graph/default_configs.py
src/physicsml/models/egnn/multi_graph/multi_graph_dataloader.py
src/physicsml/models/egnn/multi_graph/multi_graph_datamodule.py
src/physicsml/models/egnn/multi_graph/multi_graph_dataset.py
src/physicsml/models/egnn/multi_graph/multi_graph_egnn_model.py
src/physicsml/models/egnn/multi_graph/multi_graph_egnn_module.py
src/physicsml/models/egnn/multi_graph/utils.py
src/physicsml/models/egnn/ssf/__init__.py
src/physicsml/models/egnn/ssf/default_configs.py
src/physicsml/models/egnn/ssf/ssf_egnn_model.py
src/physicsml/models/egnn/ssf/ssf_egnn_module.py
src/physicsml/models/egnn/ssf/ssf_egnn_utils.py
src/physicsml/models/egnn/supervised/__init__.py
src/physicsml/models/egnn/supervised/default_configs.py
src/physicsml/models/egnn/supervised/egnn_model.py
src/physicsml/models/egnn/supervised/egnn_module.py
src/physicsml/models/mace/__init__.py
src/physicsml/models/mace/mace_prism.py
src/physicsml/models/mace/adapter/__init__.py
src/physicsml/models/mace/adapter/adapter_mace_model.py
src/physicsml/models/mace/adapter/adapter_mace_module.py
src/physicsml/models/mace/adapter/adapter_mace_utils.py
src/physicsml/models/mace/adapter/default_configs.py
src/physicsml/models/mace/mean_var/__init__.py
src/physicsml/models/mace/mean_var/default_configs.py
src/physicsml/models/mace/mean_var/mean_var_mace_model.py
src/physicsml/models/mace/mean_var/mean_var_mace_module.py
src/physicsml/models/mace/modules/__init__.py
src/physicsml/models/mace/modules/_activation.py
src/physicsml/models/mace/modules/blocks.py
src/physicsml/models/mace/modules/cg.py
src/physicsml/models/mace/modules/irreps_tools.py
src/physicsml/models/mace/modules/mace.py
src/physicsml/models/mace/modules/radial.py
src/physicsml/models/mace/modules/symmetric_contraction.py
src/physicsml/models/mace/ssf/__init__.py
src/physicsml/models/mace/ssf/default_configs.py
src/physicsml/models/mace/ssf/ssf_mace_model.py
src/physicsml/models/mace/ssf/ssf_mace_module.py
src/physicsml/models/mace/ssf/ssf_mace_utils.py
src/physicsml/models/mace/supervised/__init__.py
src/physicsml/models/mace/supervised/default_configs.py
src/physicsml/models/mace/supervised/mace_model.py
src/physicsml/models/mace/supervised/mace_module.py
src/physicsml/models/nequip/__init__.py
src/physicsml/models/nequip/nequip_prism.py
src/physicsml/models/nequip/adapter/__init__.py
src/physicsml/models/nequip/adapter/adapter_nequip_model.py
src/physicsml/models/nequip/adapter/adapter_nequip_module.py
src/physicsml/models/nequip/adapter/adapter_nequip_utils.py
src/physicsml/models/nequip/adapter/default_configs.py
src/physicsml/models/nequip/mean_var/__init__.py
src/physicsml/models/nequip/mean_var/default_configs.py
src/physicsml/models/nequip/mean_var/mean_var_nequip_model.py
src/physicsml/models/nequip/mean_var/mean_var_nequip_module.py
src/physicsml/models/nequip/modules/__init__.py
src/physicsml/models/nequip/modules/_activation.py
src/physicsml/models/nequip/modules/_gate.py
src/physicsml/models/nequip/modules/convnet_layer.py
src/physicsml/models/nequip/modules/interaction_block.py
src/physicsml/models/nequip/modules/nequip.py
src/physicsml/models/nequip/modules/radial.py
src/physicsml/models/nequip/modules/scale_shift.py
src/physicsml/models/nequip/ssf/__init__.py
src/physicsml/models/nequip/ssf/default_configs.py
src/physicsml/models/nequip/ssf/ssf_nequip_model.py
src/physicsml/models/nequip/ssf/ssf_nequip_module.py
src/physicsml/models/nequip/ssf/ssf_nequip_utils.py
src/physicsml/models/nequip/supervised/__init__.py
src/physicsml/models/nequip/supervised/default_configs.py
src/physicsml/models/nequip/supervised/nequip_model.py
src/physicsml/models/nequip/supervised/nequip_module.py
src/physicsml/models/tensor_net/__init__.py
src/physicsml/models/tensor_net/modules/__init__.py
src/physicsml/models/tensor_net/modules/embedding.py
src/physicsml/models/tensor_net/modules/interaction.py
src/physicsml/models/tensor_net/modules/output.py
src/physicsml/models/tensor_net/modules/utils.py
src/physicsml/models/tensor_net/supervised/__init__.py
src/physicsml/models/tensor_net/supervised/default_configs.py
src/physicsml/models/tensor_net/supervised/tensor_net_model.py
src/physicsml/models/tensor_net/supervised/tensor_net_module.py
src/physicsml/plugins/__init__.py
src/physicsml/plugins/ase/__init__.py
src/physicsml/plugins/ase/ase_ani.py
src/physicsml/plugins/ase/ase_graph.py
src/physicsml/plugins/ase/calculator.py
src/physicsml/plugins/ase/load.py
src/physicsml/plugins/openmm/__init__.py
src/physicsml/plugins/openmm/load.py
src/physicsml/plugins/openmm/openmm_ani.py
src/physicsml/plugins/openmm/openmm_base.py
src/physicsml/plugins/openmm/openmm_graph.py
src/physicsml/plugins/openmm/physicsml_potential.py
tests/__init__.py
tests/ase/__init__.py
tests/ase/conftest.py
tests/ase/test_ase_ani.py
tests/ase/test_ase_graph.py
tests/data/alanine-dipeptide-truncated.pdb
tests/data/ani1x.h5
tests/data/gdb9.csv
tests/data/gdb9.sdf
tests/data/ani_model/featurisation_metadata.json
tests/data/ani_model/model_config.json
tests/data/ani_model/model_artefacts/module_checkpoint.ckpt
tests/data/egnn_model/featurisation_metadata.json
tests/data/egnn_model/model_config.json
tests/data/egnn_model/model_artefacts/module_checkpoint.ckpt
tests/data/mace_model/featurisation_metadata.json
tests/data/mace_model/model_config.json
tests/data/mace_model/model_artefacts/module_checkpoint.ckpt
tests/data/nequip_model/featurisation_metadata.json
tests/data/nequip_model/model_config.json
tests/data/nequip_model/model_artefacts/module_checkpoint.ckpt
tests/openeye/__init__.py
tests/openeye/conftest.py
tests/openeye/core/__init__.py
tests/openeye/core/data/__init__.py
tests/openeye/core/data/test_datamodule.py
tests/openeye/core/data/test_graph_dataset.py
tests/openeye/core/featurisation/__init__.py
tests/openeye/core/featurisation/test_physicsml_featurisation.py
tests/openeye/integration/__init__.py
tests/openeye/integration/prism/__init__.py
tests/openeye/integration/prism/test_egnn_prism.py
tests/openeye/integration/training/__init__.py
tests/openeye/integration/training/egnn/__init__.py
tests/openeye/integration/training/egnn/test_egnn.py
tests/openmm/__init__.py
tests/openmm/conftest.py
tests/openmm/test_openmm_ani.py
tests/openmm/test_openmm_egnn.py
tests/openmm/test_openmm_mace.py
tests/openmm/test_openmm_nequip.py
tests/openmm/test_openmm_nnp_potential.py
tests/rdkit/__init__.py
tests/rdkit/conftest.py
tests/rdkit/core/__init__.py
tests/rdkit/core/data/__init__.py
tests/rdkit/core/data/test_datamodule.py
tests/rdkit/core/data/test_graph_dataset.py
tests/rdkit/core/featurisation/__init__.py
tests/rdkit/core/featurisation/test_physicsml_featurisation.py
tests/rdkit/integration/__init__.py
tests/rdkit/integration/prism/__init__.py
tests/rdkit/integration/prism/test_allegro_prism.py
tests/rdkit/integration/prism/test_ani_prism.py
tests/rdkit/integration/prism/test_egnn_prism.py
tests/rdkit/integration/prism/test_mace_prism.py
tests/rdkit/integration/prism/test_nequip_prism.py
tests/rdkit/integration/training/__init__.py
tests/rdkit/integration/training/allegro/__init__.py
tests/rdkit/integration/training/allegro/test_allegro_energy_charges.py
tests/rdkit/integration/training/allegro/test_allegro_forces.py
tests/rdkit/integration/training/allegro/test_allegro_mean_var.py
tests/rdkit/integration/training/allegro/test_allergo.py
tests/rdkit/integration/training/ani/__init__.py
tests/rdkit/integration/training/ani/test_ani.py
tests/rdkit/integration/training/ani/test_ani_ensemble.py
tests/rdkit/integration/training/ani/test_ani_forces.py
tests/rdkit/integration/training/ani/test_ani_mean_var.py
tests/rdkit/integration/training/egnn/__init__.py
tests/rdkit/integration/training/egnn/test_adapter_egnn.py
tests/rdkit/integration/training/egnn/test_egnn.py
tests/rdkit/integration/training/egnn/test_egnn_energy_charges.py
tests/rdkit/integration/training/egnn/test_egnn_forces.py
tests/rdkit/integration/training/egnn/test_egnn_mean_var.py
tests/rdkit/integration/training/egnn/test_ssf_egnn.py
tests/rdkit/integration/training/mace/__init__.py
tests/rdkit/integration/training/mace/test_adapter_mace.py
tests/rdkit/integration/training/mace/test_mace.py
tests/rdkit/integration/training/mace/test_mace_energy_charges.py
tests/rdkit/integration/training/mace/test_mace_forces.py
tests/rdkit/integration/training/mace/test_mace_mean_var.py
tests/rdkit/integration/training/mace/test_ssf_mace.py
tests/rdkit/integration/training/nequip/__init__.py
tests/rdkit/integration/training/nequip/test_adapter_nequip.py
tests/rdkit/integration/training/nequip/test_nequip.py
tests/rdkit/integration/training/nequip/test_nequip_energy_charges.py
tests/rdkit/integration/training/nequip/test_nequip_forces.py
tests/rdkit/integration/training/nequip/test_nequip_mean_var.py
tests/rdkit/integration/training/nequip/test_ssf_nequip.py
tests/rdkit/integration/training/tensor_net/__init__.py
tests/rdkit/integration/training/tensor_net/test_tensor_net.py
tests/rdkit/integration/training/tensor_net/test_tensor_net_energy_charges.py
tests/rdkit/integration/training/tensor_net/test_tensor_net_forces.py