flax
Add split and key methods to RngStream and Rngs
#5047

Open

Add split and key methods to RngStream and Rngs #5047

samanklesaria wants to merge 4 commits into google:main from samanklesaria:issues/5046

samanklesaria requested a review from

cgarciae 132 days ago

samanklesaria changed the title ~~Add split and fork methods to RngStream~~ Add split and key methods to RngStream 128 days ago

samanklesaria force pushed from f1d8d2eb to b3162ecd 128 days ago

samanklesaria force pushed from b3162ecd to 3223d3e4 128 days ago

samanklesaria force pushed from 3223d3e4 to aa6ce6af 128 days ago

Add split and fork methods to RngStream

8f8271ed

samanklesaria force pushed from aa6ce6af to b63c831d 128 days ago

samanklesaria changed the title ~~Add split and key methods to RngStream~~ Add split and key methods to RngStream and Rngs 128 days ago

samanklesaria force pushed from b63c831d to 66560425 126 days ago

Replace key property with key_

b0e18a5a

Update checkpointing guide

08044446

samanklesaria force pushed from 0e763655 to 08044446 125 days ago

cgarciae approved these changes on 2025-10-30

cgarciae added pull ready

cgarciae commented on 2025-11-12

Merge branch 'main' into issues/5046

3eea3229

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone