flax
[nnx] fix Variable overloads and add shape/dtype properties
#4049
Merged

Commits
  • [nnx] fix Variable overloads and add shape/dtype properties
    cgarciae committed 2 years ago
Loading