flax
Support linen.LogicallyPartitioned <-> nnx.Variable
#4161
Merged

Commits
  • Support linen.LogicallyPartitioned <-> nnx.Variable
    IvyZX committed 1 year ago
Loading