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