[ewalk153]

Do you see tinygrad as a useful lower level abstraction or is JAX sufficient to get pref out of AMD GPUs?

[felarof]

Tinygrad is great, but still in early stages I believe.

JAX has matured a lot over last 6 years and XLA has been around for lot longer. We believe we can extract good perf from AMD with JAX + XLA kernels.