Make ZeroGPUCompiledModel actually serializable

optimization_utils.py

@@ -45,6 +45,8 @@ class ZeroGPUCompiledModel:
             compiled_model.load_constants(self.weights.constants_map, check_full_update=True, user_managed=True)
             self.compiled_model.set(compiled_model)
         return compiled_model(*args, **kwargs)
+    def __reduce__(self):
+        return ZeroGPUCompiledModel, (self.archive_file, self.weights)
 
 
 def aoti_compile(