requirements.txt

accelerate==0.32.1
aiohappyeyeballs==2.4.0
aiohttp>=3.10.11
aiosignal==1.3.1
asttokens==2.4.1
async-timeout==4.0.3
attrs==23.2.0
beartype==0.14.1
better-abc==0.0.3
certifi==2024.7.4
charset-normalizer==3.3.2
click==8.1.7
comm==0.2.2
contourpy==1.2.1
cycler==0.12.1
datasets==2.20.0
debugpy==1.8.2
decorator==5.1.1
dill==0.3.8
docker-pycreds==0.4.0
einops==0.8.0
exceptiongroup==1.2.0
executing==2.0.1
fancy-einsum==0.0.3
filelock==3.13.1
fonttools==4.53.1
frozenlist==1.4.1
fsspec==2024.5.0
gitdb==4.0.11
GitPython==3.1.43
gmpy2==2.1.2
huggingface-hub==0.23.4
idna==3.7
importlib_metadata==8.0.0
ipykernel==6.29.5
ipython==8.26.0
ipywidgets==8.1.3
jaxtyping==0.2.31
jedi==0.19.1
Jinja2==3.1.4
jupyter_client==8.6.2
jupyter_core==5.7.2
jupyterlab_widgets==3.0.11
kiwisolver==1.4.5
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.9.1
matplotlib-inline==0.1.7
mdurl==0.1.2
mpmath==1.3.0
multidict==6.0.5
multiprocess==0.70.16
nest_asyncio==1.6.0
networkx==3.3
numpy==1.26.4
packaging==24.1
pandas==2.2.2
parso==0.8.4
pexpect==4.9.0
pickleshare==0.7.5
pillow==10.4.0
pip==24.0
platformdirs==4.2.2
prompt_toolkit==3.0.47
protobuf==5.27.2
psutil==6.0.0
ptyprocess==0.7.0
pure-eval==0.2.2
pyarrow==16.1.0
pyarrow-hotfix==0.6
Pygments==2.18.0
pygraphviz==1.13
pyparsing==3.1.2
python-dateutil==2.9.0
pytz==2024.1
PyYAML==6.0.1
pyzmq==26.0.3
regex==2024.5.15
requests==2.32.3
rich==13.7.1
safetensors==0.4.3
sentencepiece==0.2.0
sentry-sdk==2.8.0
setproctitle==1.3.3
setuptools==70.0.0
six==1.16.0
smmap==5.0.1
stack-data==0.6.2
sympy==1.12
tokenizers==0.19.1
torch==2.3.1
tornado>=6.4.2
tqdm==4.66.4
traitlets==5.14.3
transformer-lens==2.0.0
transformers==4.42.3
triton==2.3.1
typeguard==2.13.3
typing_extensions==4.11.0
tzdata==2024.1
urllib3==2.2.2
wandb==0.17.4
wcwidth==0.2.13
wheel==0.43.0
widgetsnbextension==4.0.11
xxhash==3.4.1
yarl==1.9.4
zipp==3.19.2