We need to investigate if serialization works as intended with cuda.
We need to investigate if serialization works as intended with cuda.