h5py
hyperopt
matplotlib
onnx>=1.10.2
onnxruntime>=1.10.0
overrides==3.1.0
psutil
pyyaml
ray>=1.0.0
send2trash>=1.8.0
statopt
tensorboard
tensorwatch
tqdm

[aml]
azure-ai-ml==1.5.0
azure-data-tables
azure-identity
azure-storage-blob
azureml-mlflow
ipykernel
jupyter
matplotlib
mldesigner
mlflow
lightning>=2.0.0
torchvision

[cv]
gorilla>=0.4.0
opencv-python
lightning>=2.0.0
scikit-learn
torchvision

[deepspeed]
deepspeed
mlflow

[dev]
gorilla>=0.4.0
opencv-python
lightning>=2.0.0
scikit-learn
torchvision
datasets>=2.4.0
einops
opt_einsum
tokenizers>=0.10.3
transformers==4.27.4
nbimporter
nbsphinx
nbval
pandas
pydata-sphinx-theme==0.13.1
sphinx
sphinx-book-theme
sphinx-git
sphinx-sitemap
sphinx_inline_tabs
sphinxcontrib-programoutput
sphinxcontrib-mermaid
azure-data-tables
azure-identity
azure-storage-blob
flake8>=5.0.4
pytest
tk
azure-ai-ml==1.5.0
azureml-mlflow
ipykernel
jupyter
matplotlib
mldesigner
mlflow
torchinfo
xformers
deepspeed

[docs]
nbimporter
nbsphinx
nbval
pandas
pydata-sphinx-theme==0.13.1
sphinx
sphinx-book-theme
sphinx-git
sphinx-sitemap
sphinx_inline_tabs
sphinxcontrib-programoutput
sphinxcontrib-mermaid

[flash-attn]
flash-attn

[nlp]
datasets>=2.4.0
einops
opt_einsum
tokenizers>=0.10.3
transformers==4.27.4

[tasks]
torchinfo

[tests]
azure-data-tables
azure-identity
azure-storage-blob
flake8>=5.0.4
pytest
tk

[xformers]
xformers
