You could use Halide some neural net uses I guess - like writing convolution kernels? Though (I'm not an expert on the subject) but it seems like high perf co

You could use Halide some neural net uses I guess - like writing convolution kernels? Though (I'm not an expert on the subject) but it seems like high perf convnet kernels requires assembly for specific architectures - like Scott Gray's fast Winograd kernels optimised for NVidia. Woukd be awesome if someone wrote fast conv kernels for AMD GPUs but it seems like a lot of effort