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 |
|