I am not sure that DFT-D4 has a module available on pypi, you may need to use conda or mamba. Can you also install the python extension of DFT-D4? I think you will need conda install dftd4 dftd4-python -c conda-forge per the instructions here. I remember having a similar issue before when I installed the base dftd4 package without the dftd4-python component and tried to import DFT-D4. Without the python part, the import will fail.

Also, did you run pip install torch_dftd or pip install torch-dftd? I believe the latter is what is listed on pypi and I'm not sure the installation will work with an underscore.

I am not sure that DFT-D4 has a module available on pypi, you may need to use conda or mamba.

i see. not a fan of conda or mamba (strongly prefer uv). but that's not a PR blocker ofc, esp. for an optional dep. @BowenD-UCB you wanna try running the notebook? else happy to merge as is.

PS: pip install torch-dftd and pip install torch_dftd are equivalent.

I am not sure that DFT-D4 has a module available on pypi, you may need to use conda or mamba.

i see. not a fan of conda or mamba (strongly prefer uv). but that's not a blocker PR ofc, esp. for an optional dep. @BowenD-UCB you wanna try running the notebook? else happy to merge as is.

PS: pip install torch-dftd and pip install torch_dftd are equivalent.

I had not heard of uv before but I can understand the appeal, thank you for sharing! I like the idea of a rust-based environment manager.

Hi @ajhoffman1229 and @janosh
Thanks for the PR!
Sorry for the delay on this.
I tried to install these extra dependencies but encountered installation issue (package incompatibilities) with dftd4-python
As these are optional deps I'll merge as it is now.