requirements.txt

# A list of packages that installed by default in the pytorch/pytorch docker image
# torch==1.13.0
# torchtext==0.14.0
# torchvision==0.14.0

# Transformer Lens V2
transformer-lens==1.2.2
circuitsvis==1.39.1
protobuf==3.20
torchtyping==0.1.4

# Minigrid
Minigrid @ git+https://github.com/Farama-Foundation/Minigrid.git

# torch optimizer (for lamb)
torch-optimizer==0.3.0

# tqdm
tqdm==4.64.1

# kaleido for figures
kaleido==0.2.1

# Jupyter for notebook dev
jupyter==1.0.0
jupyter-console==6.4.4
jupyter-events==0.5.0
jupyter_client==7.4.8
jupyter_core==5.1.0
jupyter_server==2.0.1
jupyter_server_terminals==0.4.2
jupyterlab-pygments==0.2.2
jupyterlab-widgets==3.0.4

# plotting
plotly==5.10.0
matplotlib==3.6.2

# pytest
pytest==7.1.3
pytest-cov==4.0.0

# wandb
wandb==0.13.7

#numpy
numpy==1.23.5

#moviepy
moviepy==1.0.3

# streamlit
streamlit==1.18.0
altair==4.2.2

# typeguard
typeguard==2.13.3

# positional encodings
positional-encodings[torch]==6.0.1

# model summaries
torchinfo==1.7.2

# dendrograms
scipy==1.10.1
scikit-learn==1.2.2

# network graphs
networkx==3.1
pyvis==0.3.2