Add sharding support to ChatSampler.#696

copybara-service[bot] wants to merge 1 commit into

maingoogle-deepmind/gemma:mainfrom

test_928907308google-deepmind/gemma:test_928907308Copy head branch name to clipboard