Add benchmark-specific pip installation option#3392

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

bvandermoon-docker-buildsAI-Hypercomputer/maxtext:bvandermoon-docker-buildsCopy head branch name to clipboard