3 ms·
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 y
by nnadams 6y ago
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?
- josevalim 6y agoSean 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!