Improve Mermaid diagrams

JuliaDiff · Mar 12, 2024 · 28aad40 · 28aad40
1 parent 7147871
commit 28aad40
Showing 1 changed file with 44 additions and 26 deletions.
diff --git a/docs/src/developer.md b/docs/src/developer.md
@@ -17,56 +17,74 @@ Advanced users are welcome to code more backends and submit pull requests!
 
 ### Forward mode 
 ```mermaid
-stateDiagram-v2
-    pushforward --> value_and_pushforward
-    value_and_pushforward --> value_and_pushforward!
-    pushforward! --> value_and_pushforward!
+flowchart LR
+    subgraph Gradient
+    gradient --> value_and_gradient
+    value_and_gradient --> value_and_gradient!
+    gradient! --> value_and_gradient!
+    end
 
+    subgraph Jacobian
     jacobian --> value_and_jacobian
     value_and_jacobian --> value_and_jacobian!
     jacobian! --> value_and_jacobian!
-    value_and_jacobian! --> value_and_pushforward!
-
-
-    gradient --> value_and_gradient
-    value_and_gradient --> value_and_gradient!
-    gradient! --> value_and_gradient!
-    value_and_gradient! --> value_and_pushforward!
+    end
 
+    subgraph Multiderivative
     multiderivative --> value_and_multiderivative
     value_and_multiderivative --> value_and_multiderivative!
     multiderivative! --> value_and_multiderivative!
-    value_and_multiderivative! --> value_and_pushforward!
-
+    end
 
+    subgraph Derivative
     derivative --> value_and_derivative
+    end
+
+    subgraph Pushforward
+    pushforward --> value_and_pushforward
+    value_and_pushforward --> value_and_pushforward!
+    pushforward! --> value_and_pushforward!
+    end
+
+    value_and_jacobian! --> value_and_pushforward!
+    value_and_gradient! --> value_and_pushforward!
+    value_and_multiderivative! --> value_and_pushforward!
     value_and_derivative --> value_and_pushforward
 ```
 
 ### Reverse mode 
 ```mermaid
-stateDiagram-v2
-    pullback --> value_and_pullback
-    value_and_pullback --> value_and_pullback!
-    pullback! --> value_and_pullback!
+flowchart LR
+    subgraph Gradient
+    gradient --> value_and_gradient
+    value_and_gradient --> value_and_gradient!
+    gradient! --> value_and_gradient!
+    end
 
+    subgraph Jacobian
     jacobian --> value_and_jacobian
     value_and_jacobian --> value_and_jacobian!
     jacobian! --> value_and_jacobian!
-    value_and_jacobian! --> value_and_pullback!
-
-
-    gradient --> value_and_gradient
-    value_and_gradient --> value_and_gradient!
-    gradient! --> value_and_gradient!
-    value_and_gradient! --> value_and_pullback!
+    end
 
+    subgraph Multiderivative
     multiderivative --> value_and_multiderivative
     value_and_multiderivative --> value_and_multiderivative!
     multiderivative! --> value_and_multiderivative!
-    value_and_multiderivative! --> value_and_pullback!
-
+    end
 
+    subgraph Derivative
     derivative --> value_and_derivative
+    end
+
+    subgraph Pullback
+    pullback --> value_and_pullback
+    value_and_pullback --> value_and_pullback!
+    pullback! --> value_and_pullback!
+    end
+
+    value_and_jacobian! --> value_and_pullback!
+    value_and_gradient! --> value_and_pullback!
+    value_and_multiderivative! --> value_and_pullback!
     value_and_derivative --> value_and_pullback
 ```