use black format all files;

remove "return state" for functions which will be executed in vmap;
recover randkey as args in mutation methods
This commit is contained in:
wls2002
2024-05-26 15:46:04 +08:00
parent 79d53ea7af
commit cf69b916af
38 changed files with 932 additions and 582 deletions

View File

@@ -0,0 +1,5 @@
import ray
ray.init(num_gpus=2)
available_resources = ray.available_resources()
print("Available resources:", available_resources)