xla
training UX: automatic generating make_train_step
#8495
Merged
Overview
Commits
11
Changes
View On GitHub

training UX: automatic generating make_train_step #8495

tengyifei merged 11 commits into master from hanq_train
qihqi
qihqi qihqi force pushed from 9f6cfac0 to 6416dd37 1 year ago
qihqi qihqi force pushed from 7ebd3466 to 0769c215 1 year ago
qihqi qihqi requested a review from tengyifei tengyifei 1 year ago
qihqi qihqi force pushed from d9c2e996 to eb3acbdf 1 year ago
qihqi qihqi force pushed from eb3acbdf to 0b0f2b59 362 days ago
qihqi qihqi requested a review from lsy323 lsy323 362 days ago
tengyifei
tengyifei requested changes on 2025-01-02
qihqi qihqi force pushed from 0b0f2b59 to dafa6b2c 339 days ago
qihqi qihqi force pushed from dafa6b2c to 075cfe52 338 days ago
qihqi qihqi requested a review from mikegre-google mikegre-google 338 days ago
qihqi qihqi requested a review from ManfeiBai ManfeiBai 338 days ago
qihqi qihqi requested a review from zpcore zpcore 338 days ago
qihqi qihqi force pushed from 075cfe52 to 56801212 338 days ago
tengyifei
qihqi qihqi removed review request from zpcore zpcore 338 days ago
qihqi qihqi removed review request from ManfeiBai ManfeiBai 338 days ago
qihqi qihqi removed review request from mikegre-google mikegre-google 338 days ago
qihqi
qihqi qihqi requested a review from tengyifei tengyifei 338 days ago
tengyifei
tengyifei commented on 2025-01-21
tengyifei
tengyifei approved these changes on 2025-01-21
qihqi qihqi force pushed from 5c8d42b1 to d9c01b1d 333 days ago
qihqi training UX: automatic generating make_train_step
ee876ff9
qihqi Add readme and docker file
89ac1824
qihqi commit Scan
dcd78e9e
qihqi checkpoint on v6e
7312ed1b
qihqi commit changes
1fc660b2
qihqi Move sharding logic inside of the model function
5149b02d
qihqi train update
3407c7a0
qihqi update from v6e examples
e23b50c9
qihqi misc fixes
20759e17
qihqi Add test and fix comment
dfc0104d
qihqi fix test fail on jax 0.5.0
9f8831a3
qihqi qihqi force pushed from d9c01b1d to 9f8831a3 333 days ago
tengyifei tengyifei merged f5b33c5d into master 333 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum