I’m relatively new to Elixir, but have been reading about translating to Nx from Numpy/PyTorch:
From Python to Elixir Machine Learning
Matrix multiplication from foundations

I was thinking that it would be helpful to be able to play around with Nx in Livebook. What I want is to be able to have the operators available, in the manner that they are available to defn functions, so that I can write in Livebook in a straightforward mathematical style (like Matlab). This is possible by having the module definition in a cell, with import Nx.Defn at the top:

Screen Shot 2024-05-30 at 2.21.44 PM1200×707 22.7 KB

But I am limited to working within this one cell. If I go outside of this cell, then I have to revert to using the pipeline style:

Screen Shot 2024-05-31 at 1.57.21 PM1200×965 37.2 KB

The way I can think to get what I want would be to wrap multiple cells in a module definition:

Screen Shot 2024-05-31 at 1.14.23 PM1200×1036 32.1 KB

Then within this context, I’m working in “Matlab mode”. I’m wondering if this is currently possible by some means in Livebook? Or if not, maybe there’s some metaprogramming means of achieving this? I’m still reading up on Elixir basics, so it’s certainly possible I’m missing some fundamental understanding about how things work…

Thanks!