[MNT] handle mps backend
for lower versions of pytorch and fix `mps…
#1931
Loading
mps backend
for lower versions of pytorch and fix `mps…
#1931