You can do that (tinygrad style) if you want.