vllm
torch
transformers
datasets
accelerate