This repo gives a good understanding of underlying code for vectors and gradient visualization
Visualize Vectors and Gradient