CUDA<- CPU interface for 3D
CUDA<- CPU interface for 3D: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Continuation of the discussion from 122. It would be good to have a way to set the Scattering3D object to CPU or CUDA modes, just like the 1D and 3D codes with cpu() and cuda() functions that mutate…
The task
Continuation of the discussion from #122. It would be good to have a way to set the Scattering3D object to CPU or CUDA modes, just like the 1D and 3D codes with cpu() and cuda() functions that mutate the object. Note that calling cuda() doesn't have to make the object move its…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.