transformers
trl
peft
accelerate
datasets

[dev]
pytest
black
isort
flake8
mypy
types-requests

[mlflow]
mlflow

[quantization]
bitsandbytes
