Change bias initialization from 'embed' to 'heads'#371

Merged

copybara-service[bot] merged 1 commit intomainfrom

fixbiassharding

Apr 16, 2026

Commits on Apr 15, 2026

Fix transformer sharding and cross-attention flash block sizes
csgoogle
committed