.codespellrc
.gitignore
.pre-commit-config.yaml
.python-version
CITATION.cff
CLAUDE.md
LICENSE
README.md
developer.md
pylintrc
pyproject.toml
tasks.py
uv.lock
.github/CODEOWNERS
.github/workflows/gpu_benchmark.yml
.github/workflows/lammps-build.yml
.github/workflows/lint.yml
.github/workflows/test.yml
assets/M3GNet.png
assets/MEGNet.png
assets/MatGL.png
assets/MxGNet.png
dev/install_dgl.sh
dev/jax_benchmark.py
dev/jax_matcalc_md.py
dev/jax_metal_md_drift.py
dev/refactor.py
dev/test_tensornet_forward_backward.py
dev/update_model_ver.py
dev/validate_tensornet.py
docs/.buildinfo
docs/404.html
docs/CLAUDE.md
docs/CNAME
docs/Gemfile
docs/Gemfile.lock
docs/LICENSE.md
docs/_config.yml
docs/changes.md
docs/conf.py
docs/developer.md
docs/genindex.html
docs/index.md
docs/matgl.apps.md
docs/matgl.data.md
docs/matgl.electrostatics.md
docs/matgl.ext.jax.md
docs/matgl.ext.md
docs/matgl.graph.md
docs/matgl.kernels.md
docs/matgl.layers.md
docs/matgl.md
docs/matgl.models.md
docs/matgl.ops.md
docs/matgl.utils.md
docs/objects.inv
docs/py-modindex.html
docs/search.html
docs/searchindex.js
docs/tutorials.md
docs/_includes/footer_custom.html
docs/_includes/nav_footer_custom.html
docs/assets/Benchmarking M3GNet Predictions of Cubic Lattice Parameters_9_1.png
docs/assets/Training a M3GNet Formation Energy Model with PyTorch Lightning_15_0.png
docs/assets/Training a MEGNet Formation Energy Model with PyTorch Lightning_15_0.png
docs/assets/Using LAMMPS with MatGL_6_1.png
docs/assets/Validation of Refitted MatGL versus Original TF M3GNet_8_0.png
docs/assets/favicon.ico
docs/tutorials/Benchmarking M3GNet Predictions of Cubic Lattice Parameters.md
docs/tutorials/CHGNet-PyG Foundation Potential.md
docs/tutorials/Combining the M3GNet Universal Potential with Property Prediction Models.md
docs/tutorials/Fine-Tuning a M3GNet Potential on the Customized Dataset with DIRECT Sampling.md
docs/tutorials/Property Predictions using MEGNet or M3GNet Models.md
docs/tutorials/Relaxations and Simulations using the M3GNet Molecular Potential.md
docs/tutorials/Relaxations and Simulations using the M3GNet Universal Potential.md
docs/tutorials/Relaxations and Simulations using the QET Universal Potential.md
docs/tutorials/Training a Formation Energy Model with PyTorch Lightning.md
docs/tutorials/Training a M3GNet Potential with PyTorch Lightning.md
docs/tutorials/Training a Multi-Fidelity MEGNet BandGap Energy Model with PyTorch Lightning.md
docs/tutorials/Training a QET Potential with PyTorch Lightning.md
docs/tutorials/Using LAMMPS with MatGL.md
examples/Benchmarking M3GNet Predictions of Cubic Lattice Parameters.ipynb
examples/CHGNet-PyG Foundation Potential.ipynb
examples/Combining the M3GNet Universal Potential with Property Prediction Models.ipynb
examples/Fine-Tuning a M3GNet Potential on the Customized Dataset with DIRECT Sampling.ipynb
examples/NVT MD with TensorNet and nvalchemi-toolkit.py
examples/Property Predictions using MEGNet or M3GNet Models.ipynb
examples/Relaxations and Simulations using the M3GNet Molecular Potential.ipynb
examples/Relaxations and Simulations using the M3GNet Universal Potential.ipynb
examples/Relaxations and Simulations using the QET Universal Potential.ipynb
examples/Training QET with Disk-Backed Datasets.ipynb
examples/Training a Formation Energy Model with PyTorch Lightning.ipynb
examples/Training a M3GNet Potential with PyTorch Lightning.ipynb
examples/Training a Multi-Fidelity MEGNet BandGap Energy Model with PyTorch Lightning.ipynb
examples/Training a QET Potential with PyTorch Lightning.ipynb
examples/Using LAMMPS with MatGL.ipynb
lammps/README.md
lammps/cmake/ML-MATGL-KOKKOS.cmake
lammps/cmake/ML-MATGL.cmake
lammps/src/KOKKOS/pair_matgl_kokkos.cpp
lammps/src/KOKKOS/pair_matgl_kokkos.h
lammps/src/ML-MATGL/export_matgl_checkpoint.py
lammps/src/ML-MATGL/pair_matgl.cpp
lammps/src/ML-MATGL/pair_matgl.h
lammps/src/ML-MATGL/patch_lammps.py
lammps/tests/check_finite_difference_pressure.py
lammps/tests/in.matgl_finite_strain
lammps/tests/in.matgl_md
lammps/tests/in.matgl_si
lammps/tests/python_reference.py
scripts/bench/bench_train_infer.py
src/matgl/__init__.py
src/matgl/cli.py
src/matgl/config.py
src/matgl/py.typed
src/matgl.egg-info/PKG-INFO
src/matgl.egg-info/SOURCES.txt
src/matgl.egg-info/dependency_links.txt
src/matgl.egg-info/entry_points.txt
src/matgl.egg-info/requires.txt
src/matgl.egg-info/scm_file_list.json
src/matgl.egg-info/scm_version.json
src/matgl.egg-info/top_level.txt
src/matgl/apps/__init__.py
src/matgl/apps/pes.py
src/matgl/data/__init__.py
src/matgl/data/transformer.py
src/matgl/electrostatics/__init__.py
src/matgl/ext/__init__.py
src/matgl/ext/_alchmtk.py
src/matgl/ext/alchmtk.py
src/matgl/ext/ase.py
src/matgl/ext/lammps.py
src/matgl/ext/pymatgen.py
src/matgl/ext/jax/__init__.py
src/matgl/ext/jax/_basis.py
src/matgl/ext/jax/_calculator.py
src/matgl/ext/jax/_convert.py
src/matgl/ext/jax/_math.py
src/matgl/ext/jax/_pad.py
src/matgl/ext/jax/_potential.py
src/matgl/ext/jax/_qet.py
src/matgl/ext/jax/_tensornet.py
src/matgl/graph/__init__.py
src/matgl/graph/_compute.py
src/matgl/graph/converters.py
src/matgl/graph/data.py
src/matgl/kernels/__init__.py
src/matgl/kernels/compose_tensor.py
src/matgl/kernels/decompose_tensor.py
src/matgl/kernels/equivariant_o3_matmul.py
src/matgl/kernels/equivariant_so3_matmul.py
src/matgl/kernels/graph_transform.py
src/matgl/kernels/tensor_norm3.py
src/matgl/kernels/tensornet_mp.py
src/matgl/kernels/tensornet_radial_mp.py
src/matgl/kernels/utils.py
src/matgl/layers/__init__.py
src/matgl/layers/_activations.py
src/matgl/layers/_atom_ref.py
src/matgl/layers/_basis.py
src/matgl/layers/_bond.py
src/matgl/layers/_core.py
src/matgl/layers/_elec_pot.py
src/matgl/layers/_embedding.py
src/matgl/layers/_embedding_warp.py
src/matgl/layers/_fast_qeq.py
src/matgl/layers/_graph_convolution.py
src/matgl/layers/_graph_convolution_warp.py
src/matgl/layers/_readout.py
src/matgl/layers/_so3.py
src/matgl/layers/_three_body.py
src/matgl/layers/_zbl.py
src/matgl/models/__init__.py
src/matgl/models/_chgnet.py
src/matgl/models/_core.py
src/matgl/models/_m3gnet.py
src/matgl/models/_megnet.py
src/matgl/models/_qet.py
src/matgl/models/_so3net.py
src/matgl/models/_tensornet.py
src/matgl/models/_wrappers.py
src/matgl/ops/__init__.py
src/matgl/ops/compose_tensor.py
src/matgl/ops/decompose_tensor.py
src/matgl/ops/equivariant_o3_matmul.py
src/matgl/ops/equivariant_so3_matmul.py
src/matgl/ops/graph_transform.py
src/matgl/ops/tensor_norm3.py
src/matgl/ops/tensornet_mp.py
src/matgl/ops/tensornet_radial_mp.py
src/matgl/utils/__init__.py
src/matgl/utils/callbacks.py
src/matgl/utils/constants.py
src/matgl/utils/cutoff.py
src/matgl/utils/io.py
src/matgl/utils/maths.py
src/matgl/utils/sb_roots.npy
src/matgl/utils/so3.py
src/matgl/utils/training.py
src/matgl/utils/uncertainty.py
tests/conftest.py
tests/test_cli.py
tests/test_config.py
tests/test_default_dtype.py
tests/test_nacl_fixture_smoke.py
tests/apps/__init__.py
tests/apps/test_pes.py
tests/data/__init__.py
tests/data/test_transformer.py
tests/electrostatics/__init__.py
tests/ext/__init__.py
tests/ext/test_alchmtk_geometry.py
tests/ext/test_alchmtk_neighborlist.py
tests/ext/test_ase.py
tests/ext/test_jax.py
tests/ext/test_lammps_export.py
tests/ext/test_lammps_run.py
tests/ext/test_pymatgen.py
tests/ext/test_tensornet_alchmtk.py
tests/graph/__init__.py
tests/graph/test_compute.py
tests/graph/test_data.py
tests/graph/test_large_data.py
tests/layers/__init__.py
tests/layers/test_activations.py
tests/layers/test_atom_ref.py
tests/layers/test_basis.py
tests/layers/test_bond.py
tests/layers/test_elec_pot.py
tests/layers/test_embedding.py
tests/layers/test_fast_qeq.py
tests/layers/test_graph_conv.py
tests/layers/test_graph_conv_chgnet.py
tests/layers/test_readout.py
tests/layers/test_so3.py
tests/layers/test_three_body.py
tests/layers/test_zbl.py
tests/models/__init__.py
tests/models/test_chgnet.py
tests/models/test_core.py
tests/models/test_m3gnet.py
tests/models/test_megnet.py
tests/models/test_qet.py
tests/models/test_so3net.py
tests/models/test_tensornet.py
tests/models/test_wrapper.py
tests/parity_data/gen_m3gnet_parity.py
tests/parity_data/gen_megnet_parity.py
tests/parity_data/gen_nacl_training_set.py
tests/parity_data/m3gnet_parity.pt
tests/parity_data/megnet_parity.pt
tests/parity_data/nacl_training_set.json.gz
tests/utils/__init__.py
tests/utils/test_constants.py
tests/utils/test_cutoff.py
tests/utils/test_io.py
tests/utils/test_matgl_potential_trainer.py
tests/utils/test_maths.py
tests/utils/test_so3.py
tests/utils/test_training.py
tests/utils/test_uncertainty.py