serialize_module_with_resources not found

When generating Torch MLIR, is there a way to write weights and model in separate file? My model weights are huge for Llama3 layers, and this is leading to giant mlir files.

I did some digging and it feels like there a way using serialize_module_with_resources based on google search, but I can't find it in the source code. 
Is it something deprecated?