flax
b9f016a0 - [nnx] RNN: add broadcast_rngs and state_axes APIs

Commit

1 year ago

[nnx] RNN: add broadcast_rngs and state_axes APIs

References

#4407 - [nnx] RNN: add broadcast_rngs and state_axes APIs

Author

cgarciae

cgarciae

Committer

cgarciae

cgarciae

Parents

Loading