iree-metal-compile.dot

digraph G {
  node [shape=rect];

  jax [label="JAX program in Python"];
  jaxpr [label="JAXPR"];
  pytorch [label="PyTorch program in Python"];
  fxgraph [label="torch.fx.Graph"];
  
  stablehlo [label="StableHLO semantics\nin MLIR source file.\nLooks like Fortran code"];
  section [label="Dispatch sections\nin StableHLO semantics.\nEach compiled into a kernel"];
  driver [label="A program that\nlaunches the kernels\nin StableHLO"];
  vmvx [label="Driver in VMVX\ninstructions executed\nby IREE runtime"];
  spirv [label="SPIR-V code\nof each section\nlooks like assembly code"];
  msl [label="Metal Shader Language code\nof each section\nLooks like C++"];
  vmfb [label="the VMFB container file\ngoing into the iOS app bundle"];
  
  jax -> jaxpr [label="  jax.make_jaxpr"];
  jaxpr -> stablehlo [label="  github.com/openxla/iree-jax"];
  
  pytorch -> fxgraph [label="  functorch"];
  fxgraph -> stablehlo [label="  github.com/llvm/torch_mlir"];
  
  stablehlo -> section [label="  split"];
  stablehlo -> driver -> vmvx -> vmfb;
  section -> spirv -> msl -> vmfb;

}

This gist document what I learned from @antiagainst.