Commit Graph

8 Commits

Author SHA1 Message Date
wls2002
72c9d4167a FAST! 2023-05-13 20:58:03 +08:00
wls2002
47b1a1dbb2 refactor genome.py use (C, 4) to replace (2, N, N) to represent connections
faster, faster and faster!
2023-05-12 00:57:55 +08:00
wls2002
b271a56827 update to test in servers 2023-05-10 22:33:51 +08:00
wls2002
3f37d79d06 gpu slice is very slow. fix this problem 2023-05-10 17:48:07 +08:00
wls2002
9dfa904ce5 precompile jax.random.split 2023-05-10 15:20:42 +08:00
wls2002
0fdc856f2d add function to put **all** compilation at the beginning of the execution. 2023-05-09 02:55:47 +08:00
wls2002
1f2327bbd6 clean imports and delete "create_XXX_functions" 2023-05-09 01:58:00 +08:00
wls2002
f63a0c447b create function_factory.py, use to manage functions 2023-05-09 01:49:43 +08:00