oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

Published 20 hours ago • verified publisher icon Oneflow-Inc

mps support

Open bghira opened this issue 8 months ago • 0 comments

Background and motivation

It would be nice to support other platforms than cuda, eg. MPS

API Proposal

Same as Pytorch.

API Usage

No response

Alternatives

MLX

Risks

No response

bghira avatar Jun 02 '24 09:06 bghira