tinygrad supports GPUs through PyOpenCL. LLaMa tinygrad/llama.py at master · geohot/tinygradYou like pytorch? You like micrograd? You love tinygrad! ❤️ - tinygrad/llama.py at master · geohot/tinygradhttps://github.com/geohot/tinygrad/blob/master/examples/llama.py