requirements.txt

numpy==1.26.4
scipy==1.12.0
matplotlib==3.8.3
Pillow==10.2.0
torch
torchvision==0.16
transformers==4.38.1
gradio==4.19.2
networkx==3.2.1
ipywidgets==8.1.2
requests==2.31.0
scikit-learn==1.4.1.post1
scikit-image==0.19.2
tqdm==4.66.2
protobuf==3.20.*
numba==0.59.0
sentencepiece==0.2.0
python-dotenv==1.0.1
gymnasium==0.29.1
pytorch_lightning==2.2.4
jupyter_ui_poll==0.2.2
tensorflow==2.8
ffmpeg
imageio-ffmpeg
torchlens
rsatoolbox
vibecheck
seaborn
plotly
git+https://github.com/Belval/TextRecognitionDataGenerator#egg=trdg
git+https://github.com/neuromatch/GNS-Modeling#egg=gns
git+https://github.com/neuromatch/pyBPL#egg=pybpl
git+https://github.com/neuromatch/MotorNet#egg=motornet
git+https://github.com/ctn-waterloo/sspspace@neuromatch#egg=sspspace
git+https://github.com/mitchellostrow/DSA#egg=DSA