https://hgpu.org/?p=19236
Array Languages Make Neural Networks Fast