porting pytorch model to vulkan compute shader - View it on GitHub