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

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

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

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

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

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

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

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

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

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

Use grain in gemma example

1225cad2

Added mixed precision: computation and weights dtypes

4fa56ad5

Code updates

b7dda724

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

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone