fixed ci workflows#128

copybara-service[bot] merged 4 commits intogoogle-deepmind:maingoogle-deepmind/jax_privacy:mainfrom

amyssnippet:workflowfixamyssnippet/jax_privacy:workflowfixCopy head branch name to clipboard