Hey, I think for this picture below the correct matmul would have `w[k,j]` and not `w[j,k]`:  https://mlc.ai/chapter_introduction/index.html
Activity