flax
Flax RNG guide for jax.jit: clarify rng outputs are shared but not inputs.
#4956

Merged

Flax RNG guide for jax.jit: clarify rng outputs are shared but not inputs. #4956

copybara-service merged 1 commit into main from test_808069188

copybara-service force pushed from 74698989 to d914ecc0 113 days ago

copybara-service force pushed from d914ecc0 to e3b9d187 111 days ago

Flax RNG guide for jax.jit: clarify rng outputs are shared but not in…

934bfe8c

copybara-service force pushed from e3b9d187 to 934bfe8c 111 days ago

copybara-service merged 934bfe8c into main 111 days ago

copybara-service deleted the test_808069188 branch 111 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone