torch
torchvision
torchaudio
accelerate
einops
fire
huggingface-hub
safetensors
sentencepiece
transformers
tokenizers
protobuf
requests
ruff
opencv-python

[tensorrt]
tensorrt-cu12
colored
onnx
onnxruntime
onnxruntime-gpu
onnx-graphsurgeon
polygraphy
