Port the collect_profile script from JAX to PathwaysUtils#71

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/pathways-utils:mainfrom

test_746578297AI-Hypercomputer/pathways-utils:test_746578297Copy head branch name to clipboard