torch
numba

[cpu]
pytorch

[gpu]
pytorch

[tests]
pytest
