Can Numba be used with Tensorflow?

后端 未结 2 748
梦如初夏
梦如初夏 2020-12-31 10:24

Can Numba be used to compile Python code which interfaces with Tensorflow? I.e. computations outside of the Tensorflow universe would run with Numba for speed. I have not fo

2条回答
  •  佛祖请我去吃肉
    2020-12-31 10:48

    I know that this does not directly answer you question, but it might be a good alternative. Numba is using just-in-time (JIT) Compilation. So, you can follow the instruction at the official TensorFlow documentation here on how to use JIT (but not in Numba ecosystem) in TensorFlow.

提交回复
热议问题