requirements.txt

absl-py>=2.1.0
aiohttp>=3.8.6
aiosignal>=1.3.1
ansiwrap>=0.8.4
async-timeout>=4.0.3
asynctest>=0.13.0
attrs>=24.2.0
backcall>=0.2.0
beartype>=0.14.1
certifi>=2024.8.30
charset-normalizer>=3.4.0
click>=8.1.7
comm>=0.1.4
cycler>=0.11.0
datasets>=2.13.2
debugpy>=1.7.0
decorator>=5.1.1
dill>=0.3.6
docker-pycreds>=0.4.0
einops>=0.6.1
entrypoints>=0.4
fancy-einsum>=0.0.3
fastjsonschema>=2.20.0
filelock>=3.12.2
fonttools>=4.38.0
frozenlist>=1.3.3
fsspec>=2023.1.0
gitdb>=4.0.11
GitPython>=3.1.43
huggingface-hub>=0.16.4
idna>=3.10
importlib-metadata>=6.7.0
importlib-resources>=5.12.0
ipykernel>=6.16.2
ipython>=7.34.0
ipywidgets>=8.1.5
jax>=0.2.22
jaxtyping>=0.2.12
jedi>=0.19.1
jsonschema>=4.17.3
jupyter_client>=7.4.9
jupyter_core>=4.12.0
jupyterlab_widgets>=3.0.13
kiwisolver>=1.4.5
markdown-it-py>=2.2.0
matplotlib>=3.5.3
matplotlib-inline>=0.1.6
mdurl>=0.1.2
multidict>=6.0.5
multiprocess>=0.70.14
nbclient>=0.7.4
nbformat>=5.8.0
nest-asyncio>=1.6.0
numpy>=1.21.6
nvidia-cublas-cu11>=11.10.3.66
nvidia-cuda-nvrtc-cu11>=11.7.99
nvidia-cuda-runtime-cu11>=11.7.99
nvidia-cudnn-cu11>=8.5.0.96
opt-einsum>=3.3.0
packaging>=24.0
pandas>=1.3.5
papermill>=2.4.0
parso>=0.8.4
pexpect>=4.9.0
pickleshare>=0.7.5
Pillow>=9.5.0
pkgutil_resolve_name>=1.3.10
platformdirs>=4.0.0
plotly>=5.18.0
prompt_toolkit>=3.0.48
protobuf>=4.24.4
psutil>=6.1.0
ptyprocess>=0.7.0
pyarrow>=12.0.1
Pygments>=2.17.2
pyparsing>=3.1.4
pyrsistent>=0.19.3
python-dateutil>=2.9.0.post0
pytz>=2024.2
PyYAML>=6.0.1
pyzmq>=26.2.0
regex>=2024.4.16
requests>=2.31.0
rich>=13.8.1
safetensors>=0.4.5
scipy>=1.7.3
seaborn>=0.12.2
sentry-sdk>=2.17.0
setproctitle>=1.3.3
six>=1.16.0
smmap>=5.0.1
tenacity>=8.2.3
textwrap3>=0.9.2
tokenizers>=0.13.3
torch>=1.13.1
tornado>=6.2
tqdm>=4.66.5
traitlets>=5.9.0
transformer-lens>=1.7.0
transformers>=4.30.2
typeguard>=4.1.2
typing_extensions>=4.7.1
urllib3>=2.0.7
wandb>=0.18.5
wcwidth>=0.2.13
widgetsnbextension>=4.0.13
xxhash>=3.5.0
yarl>=1.9.4
zipp>=3.15.0