Release 0.12.9

What's Changed

• Add vmap reset() regression test for nnx.metrics (#5483) by @chenkuanliao in #5491
• Validate dropout_rate in dot_product_attention by @mohsinm-dev in #5503
• Add RoPE embeddings by @samanklesaria in #5481
• implement axes_scan's broadcast semantics via dce by @mattjj in #5540
• Refactor NNX Variable to remove legacy JAX QDD dependencies. by @copybara-service[bot] in #5541
• [typing] add assertion to avoid pyrefly error by @copybara-service[bot] in #5546
• add pyrefly suppressions for jax.jit annotation change by @copybara-service[bot] in #5547
• Early exit in Pytree._setattr for data descriptors. by @copybara-service[bot] in #5549
• replace lazy_init implementation to avoid jax partial_eval by @mattjj in #5551
• Make Flax thread-safe for multi-threaded tracing. by @copybara-service[bot] in #5543
• release Flax 0.12.9 by @copybara-service[bot] in #5557

New Contributors

• @chenkuanliao made their first contribution in #5491

Full Changelog: v0.12.8...v0.12.9