new graph-pes versions introduce keys that aren't compatible with TorchSim interface #280
Description
The new graph-pes updates in 0.2.0 and beyond are not compatible with the TorchSim graph-pes interface, causing us to pin graph-pes in #264, @jla-gardner would you mind taking a look at this?