Version-specific `torch.scatter_reduce`

grimme-lab / dxtb

Efficient And Fully Differentiable Extended Tight-Binding

https://dxtb.readthedocs.io

Apache License 2.0

65 stars 10 forks source link

Closed marvinfriede closed 1 year ago

marvinfriede commented 1 year ago

Adapt torch.sactter_reduce to run for PyTorch versions 1.11.0, 1.12.0, 1.12.1, and 1.13.0.