llvm · henrytwo · Aug 26, 2022 · Aug 26, 2022 · Aug 26, 2022 · Aug 26, 2022
diff --git a/docs/ltc_backend.md b/docs/ltc_backend.md
@@ -76,6 +76,8 @@ Generated files are created in this directory, which is ignored by version contr
 
 ## Architecture
 
+![LTC Diagram](ltc_images/ltc_architecture.png)
+
 ### Tracing LTC graph
 
 The journey begins with a tensor in PyTorch on the `lazy` device, which may undergo a number of operations during its lifetime.
@@ -91,7 +93,7 @@ previously registered in `RegisterLazy.cpp`.
 Next, `LazyNativeFunctions::tanh` from `LazyNativeFunctions.cpp` is called, which triggers the creation of a `Tanh` node, which is a subclass of `TorchMlirNode` and `torch::lazy::Node`, defined in `LazyIr.h`.
 These nodes are then tracked internally by LTC as the computation graph is traced out.
 
-![Tracing Tensors](ltc_images/tracing_tensors.jpg)
+![Tracing Tensors](ltc_images/tracing_tensors.png)
 
 ### Syncing Tensors
 
@@ -107,15 +109,15 @@ creates an instance of `TorchMlirLoweringContext`. Here, the `TorchMlirNode`s ar
 Next, `TorchMlirLoweringContext::Build` is executed and the final `jit::Graph` is sent to `torch_mlir::importJitFunctionAsFuncOp` to generate MLIR using the existing infrastructure from Torch-MLIR.
 At this point, a `TorchMlirComputation` is created containing the final `mlir::FuncOp`.
 
-![Syncing Tensors](ltc_images/syncing_tensors.jpg)
+![Syncing Tensors](ltc_images/syncing_tensors.png)
 
 ### Final Compilation and Execution
 
 The `TorchMlirComputation` is sent to the vendor specific implementation of `TorchMlirBackendImpl::Compile` to be handed off to the vendor's compilation stack (if applicable).
 
 Finally, the compiled computation is sent to `TorchMlirBackendImpl::ExecuteComputation` to be executed on the vendor device, which produces some results to be send back to PyTorch.
 
-![Vendor Execution](ltc_images/vendor_execution.jpg)
+![Vendor Execution](ltc_images/vendor_execution.png)
 
 ## Implementing a custom backend
 

diff --git a/docs/ltc_images/ltc_architecture.png b/docs/ltc_images/ltc_architecture.png
diff --git a/docs/ltc_images/syncing_tensors.jpg b/docs/ltc_images/syncing_tensors.jpg
diff --git a/docs/ltc_images/syncing_tensors.png b/docs/ltc_images/syncing_tensors.png
diff --git a/docs/ltc_images/tracing_tensors.jpg b/docs/ltc_images/tracing_tensors.jpg
diff --git a/docs/ltc_images/tracing_tensors.png b/docs/ltc_images/tracing_tensors.png
diff --git a/docs/ltc_images/vendor_execution.jpg b/docs/ltc_images/vendor_execution.jpg
diff --git a/docs/ltc_images/vendor_execution.png b/docs/ltc_images/vendor_execution.png