einops
jaxtyping
schedulefree
liger-kernel

# visualization
tensorboard
flask
pandas
plotly