6 projects
nequip
NequIP is an open-source code for building E(3)-equivariant interatomic potentials.
nequip-allegro
Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic potentials.
tracksuit
None
torch-runstats
Running/online statistics for PyTorch
torch-ema
PyTorch library for computing moving averages of model parameters.
opt-einsum-fx
Einsum optimization using opt_einsum and PyTorch FX
