What's Changed
- update pypi publish by @cgarciae in #4538
- [nnx] register_variable_name refactor by @copybara-service in #4540
- added support to the accuracy metric for binary classification by @mattbahr in #4536
- [nnx] bridge Module by @cgarciae in #4542
- [nnx] copy _var_metadata by @copybara-service in #4548
- [bridge] fix unbox logic by @copybara-service in #4551
- Add
is_initializing
API by @copybara-service in #4550 - [nnx] Add specific model typing for nnx.Optimizer by @marcelroed in #4470
- Add linen metadata conversion to linx by @IvyZX in #4552
- [bridge] improve Module context by @cgarciae in #4554
- Raise error if user uses 'name' in bridge module setup by @IvyZX in #4555
- Add deprecation warning to all
nnx.State
methods by @IvyZX in #4561 - [nnx] add shard_map by @cgarciae in #4490
- Fix CI breakages from newest jax by @IvyZX in #4576
- [bridge] Set _initializing correctly and avoid return RNG states by @copybara-service in #4569
- v0.10.4 by @cgarciae in #4579
New Contributors
- @mattbahr made their first contribution in #4536
- @marcelroed made their first contribution in #4470
Full Changelog: v0.10.3...v0.10.4