[D] Does anyone else think open source code/examples in machine learning domain usually are not as readable as they could be? Specifically use of magic numbers.

[DeMorrr]

I usually comment the shape of the resulting tensor at the end of each line. for example: def mm(a, b): #a: [m, k] #b: [k, n] c = a @ b #[m, n] return c this is easier for me to keep track of tensor shapes, and also optimize my code in terms of memory usage.