Testing webgpu for building neural networks. - View it on GitHub