einops
jaxtyping
liger-kernel
rouge-score
transformers==4.51.1
# vllm==0.8.3
deepspeed==0.15.4 # curretly 0.16.2
accelerate==1.2.1
# visualization
tensorboard
flask
pandas
plotly