requirements.txt

absl-py==1.4.0
accelerate==0.20.3
aiohttp==3.8.4
aiohttp-cors==0.7.0
aiosignal==1.3.1
anyio
appdirs==1.4.4
argon2-cffi
argon2-cffi-bindings
asttokens
async-timeout
attrs==22.2.0
Babel
backcall
beautifulsoup4
black==23.3.0
blake3==0.3.1
bleach
blessed==1.20.0
boto3==1.26.153
botocore==1.29.153
Bottleneck
brotlipy==0.7.0
cached-property==1.5.2
cachetools==5.3.1
cattrs==22.2.0
certifi==2023.5.7
cffi
cfgv==3.3.1
charset-normalizer
chex==0.1.5
click==8.1.3
cloudpathlib==0.14.0
cmake==3.26.4
colorama
colorful==0.5.5
comet-ml==3.32.0
comm
commonmark==0.9.1
configobj==5.0.8
configparser==5.3.0
contourpy==1.0.7
coverage==7.2.5
cryptography
cycler
datasets==2.9.0
debugpy
decorator
defusedxml
dill==0.3.6
distlib==0.3.6
dm-tree==0.1.8
docker-pycreds==0.4.0
dulwich==0.21.5
einops==0.6.1
entrypoints
et-xmlfile==1.1.0
etils==1.3.0
everett==3.2.0
exceptiongroup==1.1.1
execnet==1.9.0
executing
fancy-einsum==0.0.3
fastjsonschema
filelock==3.12.0
fire==0.4.0
flash-attn==1.0.3.post0
flax==0.6.4
fonttools==4.39.4
frozenlist
fsspec==2023.5.0
gin-config==0.5.0
gitdb
GitPython
google-api-core==2.11.1
google-auth==2.20.0
googleapis-common-protos==1.59.1
gpustat==1.1
graphviz==0.20.1
grpcio==1.51.3
huggingface-hub==0.14.1
identify==2.5.24
idna
importlib-resources==5.12.0
iniconfig==2.0.0
ipykernel
ipython
ipython-genutils
ipywidgets
isort==5.12.0
jax==0.4.10
jaxlib==0.4.7
jaxtyping==0.2.19
jedi
Jinja2
jmespath==1.0.1
joblib
json5
jsonlines==3.1.0
jsonschema
jupyter
jupyter-console
jupyter-server
jupyter_client
jupyter_core
jupyterlab
jupyterlab-pygments
jupyterlab-widgets
jupyterlab_server
kiwisolver
lxml
Mako==1.2.4
markdown-it-py==2.2.0
MarkupSafe
matplotlib
matplotlib-inline
maturin==0.13.7
mdurl==0.1.2
mistune
mkl-fft==1.3.6
mkl-random
mkl-service==2.4.0
ml-dtypes==0.1.0
mpmath==1.3.0
msgpack==1.0.5
multidict==6.0.4
multiprocess==0.70.14
munkres==1.1.4
mypy==0.991
mypy-extensions==1.0.0
nbclassic
nbclient
nbconvert
nbformat
nest-asyncio
nodeenv==1.8.0
notebook
notebook_shim
numexpr
numpy
nvidia-ml-py==11.525.112
openai
opencensus==0.11.2
opencensus-context==0.1.3
openpyxl==3.0.10
opt-einsum==3.3.0
optax==0.1.5
orbax==0.1.7
packaging
pandas==1.5.3
pandas-stubs
pandocfilters
parso
pathos==0.3.0
pathspec==0.11.1
pathtools==0.1.2
pexpect
pickleshare
Pillow==9.5.0
platformdirs
plotly
pluggy==0.13.1
ply==3.11
pooch
pox==0.3.2
ppft==1.7.6.6
pre-commit==3.3.3
prometheus-client
promise==2.3
prompt-toolkit
protobuf==4.23.0
psutil
ptyprocess
pure-eval
py==1.11.0
py-spy==0.3.14
pyarrow==12.0.0
pyasn1==0.5.0
pyasn1-modules==0.3.0
pycparser
pycuda==2022.2.2
pydantic==1.10.9
Pygments
pyOpenSSL
pyparsing
PyQt5-sip==12.11.0
pyrsistent
PySocks
pytest==6.2.4
pytest-cov==3.0.0
pytest-forked==1.6.0
pytest-xdist==2.5.0
python-box==6.1.0
python-dateutil
pytools==2022.1.14
pytz
PyYAML
pyzmq
qtconsole
QtPy
ray==2.5.1
redis==3.5.3
regex==2023.5.5
requests
requests-toolbelt==1.0.0
responses==0.18.0
rich==12.6.0
rsa==4.9
s3transfer==0.6.1
safetensors==0.3.1
scikit-learn
scipy==1.10.0
seaborn==0.11.2
semantic-version==2.10.0
Send2Trash
sentry-sdk==1.23.1
setproctitle==1.3.2
setuptools-rust==1.6.0
shortuuid==1.0.11
simplejson==3.19.1
sip
six
sklearn==0.0.post5
smart-open==6.3.0
smmap==5.0.0
sniffio
sortedcontainers==2.4.0
soupsieve
stack-data
subprocess32==3.5.4
sympy==1.10.1
tabulate==0.8.9
tenacity
tensorstore==0.1.36
termcolor==2.3.0
terminado
threadpoolctl==3.1.0
tiktoken==0.3.3
tinycss2
tokenizers==0.13.3
toml
tomli==1.2.3
toolz==0.12.0
torch==1.13.1+cu116
tornado
tqdm
traitlets
transformer-lens==1.2.2
transformers @ git+https://github.com/huggingface/transformers.git@0c3fdccf2f271fb7c44f6ea6e9f4ee234795f2c5
triton==2.0.0.dev20221202
trl==0.4.7
typeguard==4.0.0
types-pytz
types-redis==3.5.6
types-tabulate==0.8.2
types-toml==0.10.8.6
typing_extensions==4.5.0
urllib3
virtualenv==20.21.0
wandb
watchdog==2.1.6
wcwidth
webencodings==0.5.1
websocket-client==1.2.3
websockets==10.1
widgetsnbextension
wrapt==1.15.0
wurlitzer==3.0.3
xxhash==3.2.0
yarl==1.9.2