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

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

..

__init__.py

initial commit

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

jax_playground.py

reduce the use of device_get in speciate

2023-05-08 15:58:09 +08:00

time_utils.py

generally complete, but not work well. Debug

2023-05-06 11:35:44 +08:00

xor.py

gpu slice is very slow. fix this problem

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