3 ms·
Do you see tinygrad as a useful lower level abstraction or is JAX sufficient to get pref out of AMD GPUs?
by ewalk153 2y ago
Do you see tinygrad as a useful lower level abstraction or is JAX sufficient to get pref out of AMD GPUs?
- felarof 2y agoTinygrad 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.