4.6. Descriptor DPA-2 PyTorch DP

Note

Supported backends: PyTorch PyTorch, DP DP

The DPA-2 model implementation. See https://arxiv.org/abs/2312.15492 for more details.

Training example: examples/water/dpa2/input_torch.json.