This is fantastic. I am just starting in the ML space (compile from compilers) and I love short kernels that I can use to understand things better with.