how to save weights

[raninbowlalala]

❓ Questions and Help

Hello, I using torchxa to convert model to stablehlo.
https://pytorch.org/xla/master/features/stablehlo.html#torch-export-to-stablehlo
Follow this page,
weights, stablehlo = tx.export.exported_program_to_stablehlo(exported)
print(stablehlo.mlir_module())
Can store weights and/or stablehlo object however you like
But how to store weights, I don't know. I found weights is a list.
Could you help me? Thank you!

Another question, we can save data and functions directory by using torch_xla， how can I save functions by using torchax?

[ysiraichi]

cc @tengyifei @zpcore