Dec 20, 2022
NequIP is an open-source code for building E(3)-equivariant interatomic potentials.
Nov 23, 2021
Running/online statistics for PyTorch
Nov 17, 2021
PyTorch library for computing moving averages of model parameters.
Nov 7, 2021
Einsum optimization using opt_einsum and PyTorch FX