numpy

[pytorch]
torch

[tensorflow]
tensorflow
