xla
support load and save checkpoint in torchax
#9616

Merged

support load and save checkpoint in torchax #9616

qihqi merged 2 commits into master from junjieqian/checkpoint

junjieqian requested a review from

vlad-karp 183 days ago

junjieqian marked this pull request as draft 183 days ago

junjieqian force pushed from 0ad1525e to d94e2cc1 183 days ago

checkpoint in torchax

fae903e9

junjieqian force pushed from d94e2cc1 to fae903e9 183 days ago

add missing dependency flax for testing

08f610a9

junjieqian force pushed from a2098b85 to 08f610a9 181 days ago

junjieqian marked this pull request as ready for review 181 days ago

junjieqian requested a review from

qihqi 181 days ago

qihqi approved these changes on 2025-09-10

qihqi merged 7aba922c into master 175 days ago

Reviewers

qihqi

vlad-karp

Assignees

No one assigned

Labels

None yet

Milestone

No milestone