WebGPU MNIST Training — Batch 64