But JAX also lets you just-in-time compile your own Python functions into XLA-optimized kernels using a one-function API, jit. algorithm 7 import pyopencl.
確定! 回上一頁