configparser
pillow
imageio[ffmpeg]
matplotlib
sentencepiece
importlib_resources
fire>=0.2.1
scikit-learn>=0.24.2
transformers>=5.7.0
datasets>=2.12.0
peft>=0.18.1
safetensors>=0.3.1

[all]
unitorch[deepspeed,diffusers,fastapis,megatron]

[deepspeed]
deepspeed>=0.11.0

[diffusers]
diffusers>=0.38.0
torchsde>=0.2.6

[docs]
mkdocstrings[python]>=0.18
mkdocs-material
markdown-exec
mkdocs-gen-files
mkdocs-coverage

[fastapis]
fastapi>=0.103.1
wandb
uvicorn
python-multipart
onnxruntime

[megatron]
megatron-core>=0.12.0
