mtenn.conversion_utils.visnet

Representation and Strategy implementations for the ViSNet model architecture. The underlying model that we use is the implementation in PyTorch Geometric.

Classes

EquivariantVecToScalar	Wrapper around torch_geometric.utils.scatter to use it as a Module.
ViSNet	mtenn wrapper around the PyTorch Geometric ViSNet model.