做了一些时间测试：

1. jit中的vmap一个函数不会触发重新编译 2. jax.lax.while_lop单独执行确实可以提前中断，但在vmap中的性能需要考虑。
2023-05-12 16:42:57 +08:00
parent 5e0130e109
commit 9b56f4ff73
2 changed files with 61 additions and 2 deletions
--- a/algorithms/neat/jitable_speciate.py
+++ b/algorithms/neat/jitable_speciate.py
@@ -0,0 +1,4 @@
+from jax import jit
+@jit
+def jitable_speciate():
+    pass