LICENSE
MANIFEST.in
README.md
pyproject.toml
src/metatrain/__init__.py
src/metatrain/__main__.py
src/metatrain/_version.py
src/metatrain.egg-info/PKG-INFO
src/metatrain.egg-info/SOURCES.txt
src/metatrain.egg-info/dependency_links.txt
src/metatrain.egg-info/entry_points.txt
src/metatrain.egg-info/requires.txt
src/metatrain.egg-info/top_level.txt
src/metatrain/cli/__init__.py
src/metatrain/cli/eval.py
src/metatrain/cli/export.py
src/metatrain/cli/formatter.py
src/metatrain/cli/train.py
src/metatrain/deprecated/__init__.py
src/metatrain/deprecated/pet/__init__.py
src/metatrain/deprecated/pet/default-hypers.yaml
src/metatrain/deprecated/pet/model.py
src/metatrain/deprecated/pet/schema-hypers.json
src/metatrain/deprecated/pet/trainer.py
src/metatrain/deprecated/pet/modules/__init__.py
src/metatrain/deprecated/pet/modules/analysis.py
src/metatrain/deprecated/pet/modules/data_preparation.py
src/metatrain/deprecated/pet/modules/hypers.py
src/metatrain/deprecated/pet/modules/long_range.py
src/metatrain/deprecated/pet/modules/molecule.py
src/metatrain/deprecated/pet/modules/pet.py
src/metatrain/deprecated/pet/modules/transformer.py
src/metatrain/deprecated/pet/modules/utilities.py
src/metatrain/deprecated/pet/tests/__init__.py
src/metatrain/deprecated/pet/tests/test_exported.py
src/metatrain/deprecated/pet/tests/test_functionality.py
src/metatrain/deprecated/pet/tests/test_pet_compatibility.py
src/metatrain/deprecated/pet/tests/test_torchscript.py
src/metatrain/deprecated/pet/utils/__init__.py
src/metatrain/deprecated/pet/utils/dataset_to_ase.py
src/metatrain/deprecated/pet/utils/fine_tuning.py
src/metatrain/deprecated/pet/utils/load_raw_pet_model.py
src/metatrain/deprecated/pet/utils/systems_to_batch_dict.py
src/metatrain/deprecated/pet/utils/update_hypers.py
src/metatrain/deprecated/pet/utils/update_state_dict.py
src/metatrain/experimental/__init__.py
src/metatrain/experimental/nanopet/__init__.py
src/metatrain/experimental/nanopet/default-hypers.yaml
src/metatrain/experimental/nanopet/model.py
src/metatrain/experimental/nanopet/schema-hypers.json
src/metatrain/experimental/nanopet/trainer.py
src/metatrain/experimental/nanopet/modules/__init__.py
src/metatrain/experimental/nanopet/modules/attention.py
src/metatrain/experimental/nanopet/modules/encoder.py
src/metatrain/experimental/nanopet/modules/feedforward.py
src/metatrain/experimental/nanopet/modules/nef.py
src/metatrain/experimental/nanopet/modules/radial_mask.py
src/metatrain/experimental/nanopet/modules/structures.py
src/metatrain/experimental/nanopet/modules/transformer.py
src/metatrain/experimental/nanopet/tests/__init__.py
src/metatrain/experimental/nanopet/tests/test_checkpoints.py
src/metatrain/experimental/nanopet/tests/test_continue.py
src/metatrain/experimental/nanopet/tests/test_exported.py
src/metatrain/experimental/nanopet/tests/test_functionality.py
src/metatrain/experimental/nanopet/tests/test_regression.py
src/metatrain/experimental/nanopet/tests/test_torchscript.py
src/metatrain/experimental/nanopet/tests/checkpoints/model-v1_trainer-v1.ckpt.gz
src/metatrain/gap/__init__.py
src/metatrain/gap/default-hypers.yaml
src/metatrain/gap/model.py
src/metatrain/gap/schema-hypers.json
src/metatrain/gap/trainer.py
src/metatrain/gap/tests/__init__.py
src/metatrain/gap/tests/ethanol_reduced_100.xyz
src/metatrain/gap/tests/options-gap.yaml
src/metatrain/gap/tests/test_errors.py
src/metatrain/gap/tests/test_exported.py
src/metatrain/gap/tests/test_regression.py
src/metatrain/gap/tests/test_torchscript.py
src/metatrain/pet/__init__.py
src/metatrain/pet/checkpoints.py
src/metatrain/pet/default-hypers.yaml
src/metatrain/pet/model.py
src/metatrain/pet/schema-hypers.json
src/metatrain/pet/trainer.py
src/metatrain/pet/modules/compatibility.py
src/metatrain/pet/modules/finetuning.py
src/metatrain/pet/modules/nef.py
src/metatrain/pet/modules/structures.py
src/metatrain/pet/modules/transformer.py
src/metatrain/pet/modules/utilities.py
src/metatrain/pet/tests/__init__.py
src/metatrain/pet/tests/test_autograd.py
src/metatrain/pet/tests/test_checkpoints.py
src/metatrain/pet/tests/test_continue.py
src/metatrain/pet/tests/test_exported.py
src/metatrain/pet/tests/test_finetuning.py
src/metatrain/pet/tests/test_functionality.py
src/metatrain/pet/tests/test_long_range.py
src/metatrain/pet/tests/test_pet_compatibility.py
src/metatrain/pet/tests/test_regression.py
src/metatrain/pet/tests/test_torchscript.py
src/metatrain/pet/tests/checkpoints/model-v1_trainer-v1.ckpt.gz
src/metatrain/pet/tests/checkpoints/model-v2_trainer-v1.ckpt.gz
src/metatrain/pet/tests/checkpoints/model-v3_trainer-v1.ckpt.gz
src/metatrain/pet/tests/checkpoints/model-v3_trainer-v2.ckpt.gz
src/metatrain/pet/tests/checkpoints/model-v4_trainer-v2.ckpt.gz
src/metatrain/pet/tests/checkpoints/model-v4_trainer-v3.ckpt.gz
src/metatrain/share/metatrain-completion.bash
src/metatrain/share/schema-base.json
src/metatrain/share/schema-dataset.json
src/metatrain/soap_bpnn/__init__.py
src/metatrain/soap_bpnn/checkpoints.py
src/metatrain/soap_bpnn/default-hypers.yaml
src/metatrain/soap_bpnn/model.py
src/metatrain/soap_bpnn/schema-hypers.json
src/metatrain/soap_bpnn/spherical.py
src/metatrain/soap_bpnn/trainer.py
src/metatrain/soap_bpnn/tests/__init__.py
src/metatrain/soap_bpnn/tests/test_checkpoints.py
src/metatrain/soap_bpnn/tests/test_continue.py
src/metatrain/soap_bpnn/tests/test_equivariance.py
src/metatrain/soap_bpnn/tests/test_exported.py
src/metatrain/soap_bpnn/tests/test_functionality.py
src/metatrain/soap_bpnn/tests/test_regression.py
src/metatrain/soap_bpnn/tests/test_torchscript.py
src/metatrain/soap_bpnn/tests/checkpoints/model-v1_trainer-v1.ckpt.gz
src/metatrain/soap_bpnn/tests/checkpoints/model-v2_trainer-v1.ckpt.gz
src/metatrain/soap_bpnn/tests/checkpoints/model-v2_trainer-v2.ckpt.gz
src/metatrain/soap_bpnn/tests/checkpoints/model-v3_trainer-v2.ckpt.gz
src/metatrain/utils/__init__.py
src/metatrain/utils/abc.py
src/metatrain/utils/architectures.py
src/metatrain/utils/augmentation.py
src/metatrain/utils/devices.py
src/metatrain/utils/dtype.py
src/metatrain/utils/errors.py
src/metatrain/utils/evaluate_model.py
src/metatrain/utils/external_naming.py
src/metatrain/utils/io.py
src/metatrain/utils/jsonschema.py
src/metatrain/utils/llpr.py
src/metatrain/utils/logging.py
src/metatrain/utils/long_range.py
src/metatrain/utils/loss.py
src/metatrain/utils/metadata.py
src/metatrain/utils/metrics.py
src/metatrain/utils/neighbor_lists.py
src/metatrain/utils/omegaconf.py
src/metatrain/utils/output_gradient.py
src/metatrain/utils/per_atom.py
src/metatrain/utils/scaler.py
src/metatrain/utils/sum_over_atoms.py
src/metatrain/utils/transfer.py
src/metatrain/utils/units.py
src/metatrain/utils/additive/__init__.py
src/metatrain/utils/additive/_base_composition.py
src/metatrain/utils/additive/composition.py
src/metatrain/utils/additive/old_composition.py
src/metatrain/utils/additive/remove.py
src/metatrain/utils/additive/zbl.py
src/metatrain/utils/data/__init__.py
src/metatrain/utils/data/combine_dataloaders.py
src/metatrain/utils/data/dataset.py
src/metatrain/utils/data/get_dataset.py
src/metatrain/utils/data/system_to_ase.py
src/metatrain/utils/data/target_info.py
src/metatrain/utils/data/readers/__init__.py
src/metatrain/utils/data/readers/ase.py
src/metatrain/utils/data/readers/metatensor.py
src/metatrain/utils/data/readers/readers.py
src/metatrain/utils/data/writers/__init__.py
src/metatrain/utils/data/writers/ase.py
src/metatrain/utils/data/writers/diskdataset.py
src/metatrain/utils/data/writers/metatensor.py
src/metatrain/utils/data/writers/writers.py
src/metatrain/utils/distributed/distributed_data_parallel.py
src/metatrain/utils/distributed/logging.py
src/metatrain/utils/distributed/slurm.py
src/metatrain/utils/testing/__init__.py
src/metatrain/utils/testing/checkpoints.py
src/metatrain/utils/testing/equivariance.py