Torchx: OOM (out of memory) on Windows

Issue created here:

https://github.com/elixir-nx/nx/issues/1637

Maybe someone else on Windows could verify my test? :slight_smile: