search

giotto-ai / giotto-deep

Deep learning made topological.
Other
76 stars 11 forks source link

Apple Silicon GPU #130

Open fbarfi opened 1 year ago

fbarfi commented 1 year ago

Is your feature request related to a problem? Please describe.

Does Giotto-deep work only on TPU? What about Apple Silicon GPU since PyTorch has provided an MPS solution to using Apple GPU with PyTorch? Thanks. I am not sure if this is the right place to ask this question. apologies in advance. thanks anyway.

Describe the solution you'd like

Describe alternatives you've considered

Additional context

RaphaelLilt commented 5 months ago

Giotto-deep supports Nvidia GPUs and TPUs out of the box. Adding support for Apple silicon would require some small changes. Please feel free to open a PR.