diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..25fb98b
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,212 @@
+absl-py==2.0.0
+aiobotocore==2.11.0
+aiofiles==22.1.0
+aiohttp==3.9.1
+aioitertools==0.11.0
+aiosignal==1.3.1
+aiosqlite==0.19.0
+annotated-types==0.6.0
+antlr4-python3-runtime==4.9.3
+anyio==4.2.0
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+arrow==1.3.0
+asttokens==2.4.1
+async-timeout==4.0.3
+attrs==23.2.0
+Babel==2.14.0
+backoff==2.2.1
+beautifulsoup4==4.12.2
+bitsandbytes==0.42.0
+black==23.12.1
+bleach==6.1.0
+blessed==1.20.0
+boto3==1.34.17
+botocore==1.34.17
+cachetools==5.3.2
+certifi==2023.11.17
+cffi==1.16.0
+charset-normalizer==3.3.2
+click==8.1.7
+comm==0.2.1
+contourpy==1.2.0
+croniter==1.4.1
+cycler==0.12.1
+dateutils==0.6.12
+debugpy==1.8.0
+decorator==5.1.1
+deepdiff==6.7.1
+defusedxml==0.7.1
+docker==6.1.3
+docstring-parser==0.15
+editor==1.6.5
+entrypoints==0.4
+exceptiongroup==1.2.0
+executing==2.0.1
+fastapi==0.109.0
+fastjsonschema==2.19.1
+filelock==3.13.1
+flake8==7.0.0
+fonttools==4.47.2
+fqdn==1.5.1
+frozenlist==1.4.1
+fsspec==2023.12.2
+google-auth==2.26.2
+google-auth-oauthlib==1.2.0
+grpcio==1.60.0
+h11==0.14.0
+hydra-core==1.3.2
+idna==3.6
+importlib-metadata==7.0.1
+importlib-resources==6.1.1
+iniconfig==2.0.0
+inquirer==3.2.1
+ipykernel==6.26.0
+ipython==8.17.2
+ipython-genutils==0.2.0
+ipywidgets==8.1.1
+isoduration==20.11.0
+isort==5.13.2
+jedi==0.19.1
+Jinja2==3.1.3
+jmespath==1.0.1
+joblib==1.3.2
+json5==0.9.14
+jsonargparse==4.27.2
+jsonpointer==2.4
+jsonschema==4.20.0
+jsonschema-specifications==2023.12.1
+jupyter-events==0.9.0
+jupyter-ydoc==0.2.5
+jupyter_client==7.4.9
+jupyter_core==5.7.1
+jupyter_server==2.12.4
+jupyter_server_fileid==0.9.1
+jupyter_server_terminals==0.5.1
+jupyter_server_ydoc==0.6.1
+jupyterlab==3.6.1
+jupyterlab-widgets==3.0.9
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.25.2
+kiwisolver==1.4.5
+lightning==2.1.3
+lightning-api-access==0.0.5
+lightning-cloud==0.5.57
+lightning-fabric==2.1.3
+lightning-utilities==0.10.0
+lightning_sdk==0.0.13a0
+Markdown==3.5.2
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+mccabe==0.7.0
+mdurl==0.1.2
+mistune==3.0.2
+mpmath==1.3.0
+multidict==6.0.4
+mypy-extensions==1.0.0
+nbclassic==1.0.0
+nbclient==0.9.0
+nbconvert==7.14.1
+nbformat==5.9.2
+nest-asyncio==1.5.8
+networkx==3.2.1
+notebook==6.5.6
+notebook_shim==0.2.3
+numpy==1.26.2
+oauthlib==3.2.2
+omegaconf==2.3.0
+ordered-set==4.1.0
+overrides==7.4.0
+packaging==23.2
+pandas==2.1.4
+pandocfilters==1.5.0
+parso==0.8.3
+pathspec==0.12.1
+pexpect==4.9.0
+pillow==10.2.0
+platformdirs==4.1.0
+pluggy==1.4.0
+prometheus-client==0.19.0
+prompt-toolkit==3.0.43
+protobuf==4.23.4
+psutil==5.9.7
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pycodestyle==2.11.1
+pycparser==2.21
+pydantic==2.5.3
+pydantic_core==2.14.6
+pyflakes==3.2.0
+Pygments==2.17.2
+PyJWT==2.8.0
+pyparsing==3.1.1
+pytest==7.4.4
+python-dateutil==2.8.2
+python-json-logger==2.0.7
+python-multipart==0.0.6
+pytorch-lightning==2.1.2
+pytz==2023.3.post1
+PyYAML==6.0.1
+pyzmq==24.0.1
+readchar==4.0.5
+redis==5.0.1
+referencing==0.32.1
+requests==2.31.0
+requests-oauthlib==1.3.1
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rich==13.7.0
+rpds-py==0.17.1
+rsa==4.9
+runs==1.2.0
+s3fs==2023.12.2
+s3transfer==0.10.0
+scikit-learn==1.3.2
+scipy==1.11.4
+Send2Trash==1.8.2
+six==1.16.0
+sniffio==1.3.0
+soupsieve==2.5
+stack-data==0.6.3
+starlette==0.35.1
+sympy==1.12
+tensorboard==2.15.1
+tensorboard-data-server==0.7.2
+tensorboardX==2.6.2.2
+terminado==0.18.0
+threadpoolctl==3.2.0
+tinycss2==1.2.1
+tomli==2.0.1
+torch==2.1.1+cu121
+torchaudio==2.1.1+cu121
+torchmetrics==1.3.0.post0
+torchvision==0.16.1+cu121
+tornado==6.4
+tqdm==4.66.1
+traitlets==5.14.1
+triton==2.1.0
+types-python-dateutil==2.8.19.20240106
+typeshed-client==2.4.0
+typing_extensions==4.9.0
+tzdata==2023.4
+uri-template==1.3.0
+urllib3==2.0.7
+uvicorn==0.25.0
+watermark==2.4.3
+wcwidth==0.2.13
+webcolors==1.13
+webencodings==0.5.1
+websocket-client==1.7.0
+websockets==11.0.3
+Werkzeug==3.0.1
+widgetsnbextension==4.0.9
+wrapt==1.16.0
+xmod==1.8.1
+y-py==0.6.2
+yarl==1.9.4
+ypy-websocket==0.8.4
+zipp==3.17.0