numpy
torch
tqdm
tensorboard

[dev]
pytest
