Add layer by layer hidden state testing support to forward_pass_logit_checker.py#4173

snehalv2002 wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

snehalv-dsv4-layer-analysisAI-Hypercomputer/maxtext:snehalv-dsv4-layer-analysisCopy head branch name to clipboard