Workflow to save top-k teacher logits in GCS to use in distillaiton#3193

copybara-service[bot] merged 2 commits intomainAI-Hypercomputer/maxtext:mainfrom

ajkv-teacher-top-k-distillationAI-Hypercomputer/maxtext:ajkv-teacher-top-k-distillationCopy head branch name to clipboard