accelerate
datasets
fire
gradio
mmengine
numpy
pybind11
safetensors
sentencepiece
setuptools
tiktoken
torch
transformers
tritonclient[all]