Docs: update torch export docs with modern canonical usage

[stellaraccident]

• move torch.compile support to an advanced/experimental feature category
• Show canonical "two liner" of how to compile with separate weights and save to a safetensors file
• Include iree-run-module example
• Remove any examples that include jittable
• Show both external and inlined parameter examples

See: https://discourse.llvm.org/t/how-to-separate-the-body-and-dialect-resources-from-a-module-in-torch-mlir-and-how-to-reload-it/81338/2?u=stellaraccident

[kumardeepakamd]

Did you mean torchscript.compile @stellaraccident in your comment?

[stellaraccident]

Referring to torch.compile here: https://iree.dev/guides/ml-frameworks/pytorch/#just-in-time-jit-execution

It is really just experimental right now and needs to be called out as such.

[vinayakdsci]

@stellaraccident just to be sure, I and @kumardeepakamd looked into this, and we had some questions.

1. move torch.compile support to an advanced/experimental feature category : shark_turbine.aot to shark_turbine.experimental.aot?
2. Show canonical "two liner" of how to compile with separate weights and save to a safetensors file : Okay, can figure this out
3. Include iree-run-module example : SG
4. Remove any examples that include jittable : Remove Just-in-time (JIT) execution section in https://iree.dev/guides/ml-frameworks/pytorch/ ?
5. Show both external and inlined parameter examples : Show inlined and non-inlined weights? The weight as a separate safetensor is one covered in above bullet, show the other one essentially when it is inlined?

[ScottTodd]

1. move torch.compile support to an advanced/experimental feature category : shark_turbine.aot to shark_turbine.experimental.aot?

My interpretation is that this focuses just on the documentation for the code as it already exists. Separately we should do the work to make the code more stable. (Also please someone pick up iree-org/iree-turbine#28 - we shouldn't be using the shark_turbine namespace this many months later)

[kumardeepakamd]

So, https://github.com/iree-org/iree-turbine/blob/main/iree/turbine/__init__.py still seems to use shark_turbine? Is someone working on changing that to iree_turbine? or should we take care of that?

[stellaraccident]

@vinayakdsci @kumardeepakamd

Another user having an issue with jittable. Can you please respond to them.

https://discourse.llvm.org/t/jittable-error-of-iree-turbine/81457/2?u=stellaraccident