added a requirements file

requirements.txt

@@ -0,0 +1,134 @@
+absl-py==2.0.0
+accelerate==0.24.1
+aiohttp==3.9.1
+aiosignal==1.3.1
+altair==5.2.0
+antlr4-python3-runtime==4.9.1
+appdirs==1.4.4
+asttokens==2.4.1
+async-timeout==4.0.3
+attrs==23.1.0
+beartype==0.14.1
+blinker==1.7.0
+cachetools==5.3.2
+certifi==2023.11.17
+charset-normalizer==3.3.2
+chex==0.1.85
+click==8.1.7
+colorama==0.4.6
+comm==0.2.0
+contourpy==1.2.0
+cycler==0.12.1
+datasets==2.15.0
+debugpy==1.8.0
+decorator==5.1.1
+dill==0.3.7
+dm-haiku==0.0.11
+docker-pycreds==0.4.0
+einops==0.7.0
+etils==1.5.2
+exceptiongroup==1.2.0
+executing==2.0.1
+fancy-einsum==0.0.3
+fastjsonschema==2.19.0
+filelock==3.13.1
+flax==0.7.5
+fonttools==4.46.0
+frozenlist==1.4.0
+fsspec==2023.10.0
+gitdb==4.0.11
+GitPython==3.1.40
+graphviz==0.20.1
+huggingface-hub==0.19.4
+idna==3.6
+importlib-metadata==6.8.0
+importlib-resources==6.1.1
+ipykernel==6.26.0
+ipython==8.18.0
+jax==0.4.20
+jaxlib==0.4.20
+jaxtyping==0.2.23
+jedi==0.19.1
+Jinja2==3.1.2
+jmp==0.0.4
+jsonschema==4.20.0
+jsonschema-specifications==2023.11.1
+jupyter_client==8.6.0
+jupyter_core==5.5.0
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+mdurl==0.1.2
+ml-dtypes==0.3.1
+mpmath==1.3.0
+msgpack==1.0.7
+multidict==6.0.4
+multiprocess==0.70.15
+nbformat==5.9.2
+nest-asyncio==1.5.8
+networkx==3.2.1
+numpy==1.26.2
+opt-einsum==3.3.0
+optax==0.1.7
+orbax-checkpoint==0.4.3
+packaging==23.2
+pandas==2.1.3
+parso==0.8.3
+Pillow==10.1.0
+platformdirs==4.0.0
+plotly==5.18.0
+prompt-toolkit==3.0.41
+protobuf==4.25.1
+psutil==5.9.6
+pure-eval==0.2.2
+pyarrow==14.0.1
+pyarrow-hotfix==0.6
+pydeck==0.8.1b0
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+pytz==2023.3.post1
+pywin32==306
+PyYAML==6.0.1
+pyzmq==25.1.1
+referencing==0.31.0
+regex==2023.10.3
+requests==2.31.0
+rich==13.7.0
+rpds-py==0.13.1
+safetensors==0.4.0
+scipy==1.11.4
+sentry-sdk==1.37.1
+setproctitle==1.3.3
+six==1.16.0
+smmap==5.0.1
+stack-data==0.6.3
+streamlit==1.29.0
+sympy==1.12
+tabulate==0.9.0
+tenacity==8.2.3
+tensorstore==0.1.50
+tokenizers==0.15.0
+toml==0.10.2
+toolz==0.12.0
+torch==2.1.1
+tornado==6.3.3
+tqdm==4.66.1
+tracr==1.0.0
+traitlets==5.13.0
+transformer-lens==1.10.0
+transformers==4.35.2
+typeguard==2.13.3
+typing_extensions==4.8.0
+tzdata==2023.3
+tzlocal==5.2
+urllib3==2.1.0
+validators==0.22.0
+wandb==0.16.0
+watchdog==3.0.0
+wcwidth==0.2.12
+xxhash==3.4.1
+yarl==1.9.3
+zipp==3.17.0