LICENSE
README.md
pyproject.toml
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/top_level.txt
src/matgl/apps/__init__.py
src/matgl/apps/_pes_dgl.py
src/matgl/apps/_pes_pyg.py
src/matgl/apps/pes.py
src/matgl/data/__init__.py
src/matgl/data/transformer.py
src/matgl/electrostatics/__init__.py
src/matgl/electrostatics/_elec_pot_dgl.py
src/matgl/electrostatics/_fast_qeq_dgl.py
src/matgl/ext/__init__.py
src/matgl/ext/_alchmtk.py
src/matgl/ext/_ase_dgl.py
src/matgl/ext/_ase_pyg.py
src/matgl/ext/_pymatgen_dgl.py
src/matgl/ext/_pymatgen_pyg.py
src/matgl/ext/ase.py
src/matgl/ext/pymatgen.py
src/matgl/graph/__init__.py
src/matgl/graph/_compute_dgl.py
src/matgl/graph/_compute_pyg.py
src/matgl/graph/_converters_dgl.py
src/matgl/graph/_converters_pyg.py
src/matgl/graph/_data_dgl.py
src/matgl/graph/_data_pyg.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_dgl.py
src/matgl/layers/_atom_ref_pyg.py
src/matgl/layers/_basis.py
src/matgl/layers/_bond.py
src/matgl/layers/_core.py
src/matgl/layers/_core_dgl.py
src/matgl/layers/_embedding_dgl.py
src/matgl/layers/_embedding_pyg.py
src/matgl/layers/_embedding_warp.py
src/matgl/layers/_graph_convolution_dgl.py
src/matgl/layers/_graph_convolution_pyg.py
src/matgl/layers/_graph_convolution_warp.py
src/matgl/layers/_norm.py
src/matgl/layers/_readout_dgl.py
src/matgl/layers/_readout_pyg.py
src/matgl/layers/_readout_torch.py
src/matgl/layers/_so3.py
src/matgl/layers/_three_body.py
src/matgl/layers/_zbl_dgl.py
src/matgl/layers/_zbl_pyg.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_dgl.py
src/matgl/models/_so3net.py
src/matgl/models/_tensornet_dgl.py
src/matgl/models/_tensornet_pyg.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/_training_dgl.py
src/matgl/utils/_training_pyg.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
tests/test_cli.py
tests/test_config.py
tests/test_default_dtype.py
tests/test_integration.py