Tweeted By @jeremyphoward
The growing popularity of JAX could be a disaster for @NvidiaAI unless they fix the poor support for GPUs in XLA.
— Jeremy Howard (@jeremyphoward) February 23, 2020
Currently JAX only runs well on Google's TPUs. https://t.co/NQHxqVAGR0