wls2002 9b56f4ff73 做了一些时间测试： ...

1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断，但在vmap中的性能需要考虑。

2023-05-12 16:42:57 +08:00

..

__pycache__

initial commit in branch developing

2023-05-11 19:04:56 +08:00

__init__.py

initial commit

2023-05-05 14:19:13 +08:00

function_tests.py

refactor genome.py use (C, 4) to replace (2, N, N) to represent connections

2023-05-12 00:57:55 +08:00

jax_playground.py

做了一些时间测试：

2023-05-12 16:42:57 +08:00

time_utils.py

generally complete, but not work well. Debug

2023-05-06 11:35:44 +08:00

xor.py

refactor genome.py use (C, 4) to replace (2, N, N) to represent connections

2023-05-12 00:57:55 +08:00