[bcardarella]

Nx solves all of the issues you're citing. That is how you do ML in Elixir