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 98 days ago

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

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

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

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

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

Use grain in gemma example

549a4486

Added mixed precision: computation and weights dtypes

e17d2fec

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

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

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

Code updates

1049e5f1

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

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone