matplotlib
pillow
torch

[dev]
