[outlog]

amazing work ♥!

any idea on M1 gpu support? (know it's mostly an upstream issue)

excited about "real"/intensive ML workflows locally..

[josevalim]

You 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.