flax
2c2a7961 - make params_with_axes() work when params_axes is not mutable

Commit

3 years ago

make params_with_axes() work when params_axes is not mutable PiperOrigin-RevId: 424138995

References

Author

a-googler

Committer

a-googler

Parents