numpy
torch
pandas
solara
einops
transformers==4.34.1