flax
[nnx] fix Rngs dtype check
#4820
Merged

[nnx] fix Rngs dtype check #4820

copybara-service merged 1 commit into main from nnx-fix-rng-dtype-check
cgarciae
cgarciae [nnx] fix Rngs dtype check
164c6be2
IvyZX
IvyZX approved these changes on 2025-07-16
IvyZX IvyZX added pull ready
copybara-service copybara-service merged 8447c032 into main 1 year ago
copybara-service copybara-service deleted the nnx-fix-rng-dtype-check branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone