pytorch

b38b39c4 - Revert "memory viz: Add colors for categories and a legend (#90587)"