google/flax v0.10.4

Release 0.10.4

on GitHub

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