[gpm]

> Riff doesn't track which GPU you use at this time, so it can't track that which build of `pytorch` to use, sorry. It might be something we make it aware of if we have enough evidence it would be helpful.

Would it be possible to do something like make it depend on a cargo feature flag, or something similar?

It's probably better that this isn't automatic, since the correct choice of device isn't always obvious just based on what is available.