torch
bitsandbytes
deepspeed
einops
triton
rwkv-fla
rwkv
transformers
GPUtil
plotly
lightning
datasets
miditok
wandb