It acutally learns (again)

After some more pain with the loss function, buffer sizes and shape mismatches I got a neural network running on the gpu. The GPU is about 10x faster than the CPU version although some hyperparameters can affect this result.

Now I want to build a small wasm interface and build a small web page around that to allow anyone to run a local neural net in the browser.