[Question] Specifying the device to run on

[erlebach]

I am on a mac, and cannot find where one specifies the device to run on, which in my case is 'mps'. Should I use the standard approach using PyTorch, or is there a more idiomatic approach in zeta? I assume that the library is sufficiently general to run on Nvidia GPUs, ARM GPUs, and the CPU. Thanks.

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.

[github-actions]

Hello there, thank you for opening an Issue ! 🙏🏻 The team was notified and they will get back to you asap.

[kyegomez]

@erlebach cpu works fine, torch doesn't support m hardware