torch
numpy

[bench]
matplotlib
tqdm

[progress]
tqdm

[test]
pytest

[triton]
triton
