2 ms·PyTorch MPS is about 10x faster per the README.md.by throwaway314155 9mo agoPyTorch MPS is about 10x faster per the README.md.antirez 9mo agoI cut the difference in speed by half by taking the activations on the GPU. Time to sleep but will continue tomorrow.Numerlor 9mo agoHave you tried e.g. Mojo that can vectorize/do SIMD without having to do intrinsics everywhere?