top | item 41635953

(no title)

ewalk153 | 1 year ago

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

discuss

order

felarof|1 year ago

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.