5 ms·
amazing work ♥! any idea on M1 gpu support? (know it's mostly an upstream issue) excited about "real"/intensive ML workflows locally..
by outlog 5y ago
amazing work ♥!
any idea on M1 gpu support? (know it's mostly an upstream issue)
excited about "real"/intensive ML workflows locally..
- josevalim 5y agoYou are correct. Currently we are waiting for the issue to be tackled upstream (both on LibTorch and Google's XLA). Someone could also implement a specific Nx-backend for Apple. It should actually be fun but not high priority at the moment.