fchollet介绍JAX实现快速且支持TPU

这些实现非常快——当使用JAX运行时，你会获得最先进的性能。而且它们都可以在TPU上运行。