Google TPUs with Mojo language?
Sup,
I'm creating a resource-intensive AI (at least 3-4x more resource-intensive than LLMs), so I'm considering a stack which would allow me fast computation cheap (while making dev relatively easy).
Would using JAX and TPUs not work with Mojo? As far as I see, using TPUs is the absolute fastest we can get for deep learning. (or is it? Would MAX inference engine be faster?)
Thanks everyone!
I'm creating a resource-intensive AI (at least 3-4x more resource-intensive than LLMs), so I'm considering a stack which would allow me fast computation cheap (while making dev relatively easy).
Would using JAX and TPUs not work with Mojo? As far as I see, using TPUs is the absolute fastest we can get for deep learning. (or is it? Would MAX inference engine be faster?)
Thanks everyone!
