torch
numpy

[cupy]
cupy
