mtenn.conversion_utils.schnet

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

Classes

MemoizedRadiusInteractionGraph	Memoized version of the PyG RadiusInteractionGraph.
SchNet	mtenn wrapper around the PyTorch Geometric SchNet model.