Numba可以和Tensorflow一起使用吗？

Question

Numba可以用来编译与Tensorflow接口的Python代码吗？即Tensorflow宇宙之外的计算将与Numba一起运行以获得速度。我没有找到任何有关如何执行此操作的资源。

Answer 1

我知道这并没有直接回答你的问题，但它可能是一个很好的选择。 Numba正在使用即时（JIT）编译。因此，您可以按照TensorFlow官方文档here中有关如何在TensorFlow中使用JIT（但不在Numba生态系统中）的说明进行操作。