Learning ML by reading tinygrad - Part 2