由于使用XLA(一种加速线性代数计算的编译器)将Python和JAX代码JIT编译成优化的内核, ... T).block_until_ready() # 2. measure JAX compilation time
確定! 回上一頁