Skip to content

Improve Gemma example #4740

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
cgarciae opened this issue May 7, 2025 · 0 comments
Open

Improve Gemma example #4740

cgarciae opened this issue May 7, 2025 · 0 comments
Assignees
@vfdev-5

Comments

@cgarciae
Copy link
Collaborator

cgarciae commented May 7, 2025 •
edited
Loading

  • Add training code from l1mb_nnx
  • Add support distributed training via sharding
  • Add support for multiple samples per sequence as in lm1b by modifying the attention layer.
  • Replace tfds with grain.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@vfdev-5 vfdev-5

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@vfdev-5 @cgarciae