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