numpy
scipy
matplotlib
torch
jax

[llm]
transformers
datasets
tqdm
accelerate

[nb]
jupyter
