News for hackers. Stuff that matters.

A comment on Nx: Multi-dimensional tensors Elixir lib with multi-staged compilation (CPU/GPU). Parent comment

Re:Nx: Multi-dimensional tensors Elixir lib with multi-stage... by nnadams on Wednesday February 17, 2021 @02:38PM (#26170562)
I'm excited to try this out, especially after seeing the XLA support.

Where do you see Nx going in the future? I saw there's an MNIST example in the repo. Do you plan including higher-level features for neural networks, similar to Keras/PyTorch?

[ Parent | on HN ]
Re:Nx: Multi-dimensional tensors Elixir lib with multi-stage... by josevalim on Wednesday February 17, 2021 @02:50PM (#26170747)
Sean is looking into higher-level features for NN - it will likely be a separate library on top of Nx. For general direction, I briefly discuss that at the end of the article above. :) If something is unclear, please let me know!
[ Parent | on HN ]