xla
Backport Fix preserve_rng_state for activation checkpointing (#4690)
#6788

Merged

Backport Fix preserve_rng_state for activation checkpointing (#4690) #6788

JackCaoG merged 1 commit into r2.3 from JackCaoG/backport_oerserve_rng

Fix preserve_rng_state for activation checkpointing (#4690)

49efb2b6

JackCaoG requested a review from

lsy323 1 year ago

lsy323 approved these changes on 2024-03-22

JackCaoG merged 1d6558a3 into r2.3 1 year ago

Reviewers

lsy323

Assignees

No one assigned

Labels

None yet

Milestone

No milestone