numpy
torch
dataclasses