Update README.md

sail-sg · Jan 4, 2024 · 0818381 · 0818381
1 parent 8be3d1a
commit 0818381
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,7 @@
 # Automatic Functional Differentiation in JAX
 
+<img src='./autofd.svg' width='200'>
+
 **TLDR:** We implemented automatic **functional differentiation** (as in variational calculus) in JAX, one can do `g=jax.grad(F)(f)` to get the derivative of the functional `F` at function `f`, where `g` is itself a callable python function.
 
 ## Installation and Usage