Updated viz_manager.py

[ajinkya-kulkarni]

Modified the nested for loops that iterate over the image grid to use numpy vectorization instead, by reshaping the images and titles into 2D arrays, and then iterating over the first dimension of those arrays. This allows for faster iteration and better performance when dealing with large sets of images.

[CLAassistant]

All committers have signed the CLA.

[jwmueller]

@ajinkya-kulkarni Thanks for your awesome contribution!

Could you please sign the Contributor License Agreement linked above? Thank you!

[sanjanag]

Hi @ajinkya-kulkarni ! Thank you for the contribution. Could you post some results on performance gains and how the visualizations look like after this change?
Also, the PR is failing on some CI checks. Could you please fix these as well? These checks are important for testing and code maintenance. You can check the development guide for further details.
Feel free to let me know if you need anymore details.

[ajinkya-kulkarni]

I will look into why is it failing.

[jwmueller]

@ajinkya-kulkarni Any chance you can address the above comments from @sanjanag so we can get this PR in? Thanks!

[jwmueller]

@ajinkya-kulkarni Any updates on this PR?