visualize (pypi): useful visualization function
Keywords: computer|vision|attention|attention-map|attention-mechanism|data-visualization|deep-learning|transformer|vision-mlp|vision-transformer|visualization