PR #4913 Updates in gemma example

Updates in gemma example #4913

vfdev-5 wants to merge 3 commits into main from gemma-example-using-grain

vfdev-5 changed the title ~~Updates in gemma example: grain dataflow, simplified sharding rules~~ Updates in gemma example 187 days ago

vfdev-5 force pushed from 5ad27b52 to 3beac991 172 days ago

vfdev-5 force pushed from 3beac991 to aa1683de 165 days ago

vfdev-5 force pushed from aa1683de to e4e8597b 151 days ago

vfdev-5 force pushed from e4e8597b to e2d5e40d 143 days ago

vfdev-5 force pushed from e2d5e40d to cc785f60 143 days ago

vfdev-5 force pushed from cc785f60 to a4a7c907 137 days ago

vfdev-5 force pushed from a4a7c907 to f6b6f7b8 131 days ago

vfdev-5 force pushed from f6b6f7b8 to 79ec9708 131 days ago

vfdev-5 force pushed from 79ec9708 to 1049e5f1 130 days ago

vfdev-5 force pushed from 1049e5f1 to b7dda724 78 days ago

vfdev-5 force pushed from b7dda724 to 1b42250a 26 days ago

vfdev-5 commented on 2026-03-11

vfdev-5 force pushed from 1b42250a to 634512ba 19 days ago

vfdev-5 force pushed from 634512ba to 9c711dd9 19 days ago

vfdev-5 force pushed from 9c711dd9 to 9b2b5ddb 19 days ago

vfdev-5 marked this pull request as ready for review 19 days ago

Use grain in gemma example

199e6530

Added mixed precision: computation and weights dtypes

cdf3b0c3

vfdev-5 force pushed from 9b2b5ddb to e22ece9a 12 days ago

cgarciae approved these changes on 2026-03-25

cgarciae added pull ready

vfdev-5 force pushed from e22ece9a to a860938d 12 days ago

IvyZX requested changes on 2026-03-25

More Code updates

5260a1ed

vfdev-5 force pushed from a860938d to 5260a1ed 11 days ago

vfdev-5 requested a review from

IvyZX 11 days ago

cgarciae approved these changes on 2026-04-03

Reviewers

cgarciae

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone