Commit Graph

4 Commits

Author SHA1 Message Date
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