Fix docker build and upload with pip install#3448

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

docker_buildAI-Hypercomputer/maxtext:docker_buildCopy head branch name to clipboard