v0.7.5

[levskaya]

What's Changed

• Add method-to-model section to Haiku migration guide by @IvyZX in Add method-to-model section to Haiku migration guide #3277
• updated haiku guide with new JAX RNG api by @chiamp in updated haiku guide with new JAX RNG api #3343
• changed resnet v1 to v1.5 by @chiamp in changed resnet v1 to v1.5 #3344
• updated flax_basics by @chiamp in updated flax_basics #3342
• Add find methods and magic methods for Cursor API by @chiamp in Add find methods and magic methods for Cursor API #3306
• fix DeprecationWarnings by @chiamp in fix DeprecationWarnings #3352
• Make checkpoint guide path absolute by @IvyZX in Make checkpoint guide path absolute #3358
• use jax.Array type for rng keys by @chiamp in use jax.Array type for rng keys #3354
• Disable pyink by @cgarciae in Disable pyink #3356
• removed DeprecationWarning filter by @chiamp in removed DeprecationWarning filter #3359
• Don't propagate default args in Tabulate by @cgarciae in Don't propagate default args in Tabulate #3357
• added spectral norm by @chiamp in added spectral norm #3335
• fixed bind-unbind bug by @chiamp in fixed bind-unbind bug #3365
• Add MaxText open source LLM to index.rst by @8bitmp3 in Add MaxText open source LLM to index.rst #3368
• fixed typo by @chiamp in fixed typo #3367
• Add fp8 custom op and unit test by @wenscarl in Add fp8 custom op and unit test #3322
• fixed scope typing by @chiamp in fixed scope typing #3371
• Trailing whitespace fixes. by @levskaya in Trailing whitespace fixes. #3373
• Make fp8 ops use explicit broadcasting. by @levskaya in Make fp8 ops use explicit broadcasting. #3374
• Conditonal contraints for numpy and clu by @cgarciae in Conditonal contraints for numpy and clu #3394
• add truncated_normal to initializers by @levskaya in add truncated_normal to initializers #3401
• fix HEAD by @chiamp in fix HEAD #3404
• updated imagenet readme by @chiamp in updated imagenet readme #3383
• Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics by @8bitmp3 in Add Flax FAQ - how to search, how to take the derivative w.r.t. a hidden layer, remat_scan vs scan(remat), recommended training libraries/metrics #3267
• added rmsnorm to api docs by @chiamp in added rmsnorm to api docs #3406
• updated docs by @chiamp in updated docs #3370
• fix HEAD by @chiamp in fix HEAD #3408
• added weightnorm layer by @chiamp in added weightnorm layer #3405
• added attention refactor to changelog by @chiamp in added attention refactor to changelog #3412
• added dropout arg to MultiHeadDotProductAttention by @chiamp in added dropout arg to MultiHeadDotProductAttention #3384
• fix spectralnorm layer by @chiamp in fix spectralnorm layer #3403
• remove pdf target by @cgarciae in remove pdf target #3415
• added import precommit hook by @chiamp in added import precommit hook #3410
• fixed GRU docstring by @chiamp in fixed GRU docstring #3419
• Replaces pjit with jit in spmd.py by @copybara-service in Replaces pjit with jit in spmd.py #3421
• Ignore transient chex deprecationwarning. by @levskaya in Ignore transient chex deprecationwarning. #3427
• Remove transformers dependency from docs by @cgarciae in Remove transformers dependency from docs #3431
• updated pytorch upgrade guide by @chiamp in updated pytorch upgrade guide #3432
• Simplify abstract rng creation in param shape check. by @levskaya in Simplify abstract rng creation in param shape check. #3429
• [NVIDIA] Update the FP8 support by @kaixih in [NVIDIA] Update the FP8 support #3435
• fixed mypy errors at HEAD by @chiamp in fixed mypy errors at HEAD #3440
• added has_improved field to EarlyStopping by @chiamp in added has_improved field to EarlyStopping #3385
• fully deprecated old RNN api by @chiamp in fully deprecated old RNN api #3425
• updated lm1b example with jit by @chiamp in updated lm1b example with jit #3302
• added MGU class by @chiamp in added MGU class #3418
• Make Flax Basics visible by @8bitmp3 in Make Flax Basics visible #3443
• update for release v0.7.5 by @levskaya in update for release v0.7.5 #3444

New Contributors

• @wenscarl made their first contribution in Add fp8 custom op and unit test #3322
• @kaixih made their first contribution in [NVIDIA] Update the FP8 support #3435

Full Changelog: v0.7.4...v0.7.5

---

This discussion was created from the release v0.7.5.