can't call as_array() on a torch_float32() tensor on "mps" (M1 Mac Studio) GPU

Cross-posted here:

Please familiarize yourself with our crossposting policy