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