numpy
lark-parser
torch
dataclasses
einops
