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.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.py
src/matgl/electrostatics/_fast_qeq.py
src/matgl/ext/__init__.py
src/matgl/ext/_alchmtk.py
src/matgl/ext/_ase.py
src/matgl/ext/_pymatgen.py
src/matgl/ext/_tensornet_alchmtk.py
src/matgl/ext/alchmtk.py
src/matgl/ext/ase.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/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/_embedding.py
src/matgl/layers/_embedding_warp.py
src/matgl/layers/_grace.py
src/matgl/layers/_graph_convolution.py
src/matgl/layers/_graph_convolution_warp.py
src/matgl/layers/_readout.py
src/matgl/layers/_readout_torch.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/_grace.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/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_nacl_fixture_smoke.py