When I was learning Pytorch I saw this video:
So knowing that Pytorch can basically do Numpy on the GPU,
I thought it would be interesting to simulate the agents using Pytorch.
And this is my result:
It's not quit the same result as Sebastian Lague had,
probably because there are quite a few parameters that change the outcome massively.
But this is satisfying enough.