Add T5gemma2 - #145

Draft

jrabary wants to merge 7 commits into

jax-ml:mainfrom

jrabary:t5gemma2

Commits on Jan 26, 2026

init t5gemma2 model
jrabary
committed
4c004ae
Add T5Gemma2 model implementation
jrabary
committed
7a91bfc

Commits on Jan 27, 2026

Refactor T5Gemma2 model components and add parameter handling and test code
jrabary
committed
02331b3
Fix formating
jrabary
committed
4b04264

Commits on Jan 30, 2026

Merge branch 'main' into t5gemma2
jenriver
authored
f6a17db

Commits on Feb 2, 2026

Add sharding configuration classes, update model parameters and add test
jrabary
committed
e534b26
Merge branch 't5gemma2' of https://github.com/jrabary/bonsai into t5gemma2
jrabary
committed
08147bf