flax
[nnx] RNN: add broadcast_rngs and state_axes APIs
#4407
Merged
Overview
Commits
1
Changes
View On GitHub
Commits
  • [nnx] RNN: add broadcast_rngs and state_axes APIs
    cgarciae committed 1 year ago
FAQ Terms Privacy Refunds Impressum
Loading