cupy
torch
pytest
