diff --git a/docs/assets/use_cases/node_representation_learning/context_proba_v3.png b/docs/assets/use_cases/node_representation_learning/context_proba_v3.png
new file mode 100644
index 000000000..c2563e49d
Binary files /dev/null and b/docs/assets/use_cases/node_representation_learning/context_proba_v3.png differ
diff --git a/docs/assets/use_cases/node_representation_learning/sage_layer_eqn_v3.png b/docs/assets/use_cases/node_representation_learning/sage_layer_eqn_v3.png
new file mode 100644
index 000000000..8d7992dff
Binary files /dev/null and b/docs/assets/use_cases/node_representation_learning/sage_layer_eqn_v3.png differ