transformers
torch
huggingface_hub
pynvml
rwkv
Ninja