We get an error 137 (probably excessive memory usage?). So I will try to use jit in pre_slide_vmap function. - [x] Check on Converging PC - [x] Change **add** to "to_batch" suggested by Kevin - [x] detach pickle / dill from the init_vec and modify model parameters - [x] test init_vec - [x] goal: having an executable solution to merge in main