wls2002 3f37d79d06 gpu slice is very slow. fix this problem

2023-05-10 17:48:07 +08:00

..

__pycache__

initial commit

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

genome

add function to put **all** compilation at the beginning of the execution.

2023-05-09 02:55:47 +08:00

__init__.py

initial commit

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

function_factory.py

gpu slice is very slow. fix this problem

2023-05-10 17:48:07 +08:00

pipeline.py

gpu slice is very slow. fix this problem

2023-05-10 17:48:07 +08:00

species.py

precompile jax.random.split

2023-05-10 15:20:42 +08:00