Files
tensorneat-mend/algorithms/neat/jitable_speciate.py
wls2002 9b56f4ff73 做了一些时间测试:
1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断,但在vmap中的性能需要考虑。
2023-05-12 16:42:57 +08:00

4 lines
57 B
Python

from jax import jit
@jit
def jitable_speciate():
pass