search

Oneflow-Inc / oneflow

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
http://www.oneflow.org
Apache License 2.0
5.87k stars 667 forks source link

mps support #10534

Open bghira opened 3 months ago

bghira commented 3 months ago

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