flax

342a5e83 - [nnx] fix ToNNX linen_attributes update