Hey @ds604, you can do use Nx.Defn.Kernel at the top of your notebook, this is going to replace default imports, such as +, as it would within defn. We don’t recommend this in the docs or elsewhere, but it can be neat for experimentation, like in your case : )
