torch
transformers>=4.36.0
colossalai==0.3.6
click
loguru
fabric
cornstarch
grpcio
pulp

[dev]
torch>=2.1.0
ruff
black>=23.0
isort>=5.12
pytest
pytest-mock
pytest-grpc
grpcio-tools
datasets
