Well done! I would be crazy if I dived into this having never studied Rust! But it looks very interesting so I might do that! I have almost finished AK’s lecture. Have you made the matmul faster yet? That would seem a priority?
Hey! Yes please dive in :) It can be a starting point for learning Rust ;)
MatMul is fine, the bottleneck is LayerNorm + Softmax gradients right now. Haven't been able to optimize them well yet.
6 comments
[ 1.4 ms ] story [ 30.8 ms ] threadi7-13700H, 2400 Mhz, 14 Core(s), 20 Logical Processor(s). It is using 12 CPUs which is good. And 160Mb memory.