einops
torch
transformer-lens
matplotlib
tqdm
numpy
pytest
plotly
ipywidgets
