This repository contains the code for visualization of various deep learning optimization algorithms. PyTorch is used for automatic differentiation and implementing the various optimizers.
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |
| 2D Convex Surface | 2D Non-Convex Surface | 3D Surface with Saddle Point |
|---|---|---|
![]() |
![]() |
![]() |











