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