In the book we lean on the EXLA and I haven’t had issues with it either on Linux or Mac when configuring the compiler for :host. I have been meaning to try out GitHub - ausimian/emily: Elixir bindings and Nx backend for Apple MLX · GitHub to make use of MLX on my Apple hardware, but haven’t had the chance yet. If I had any CUDA hardware I could probably share your pain 
.
