┌────────────┐
│Optimizer   │
├────────────┤
└────────────┘
     |        
     |        
   ┌─────┐    
   │Torch│    
   ├─────┤    
   └─────┘