Commits ⋅ google/flax

Merge pull request #2231 from cgarciae:fix-notebooks

main 5b4ef40

Unable to find commit in GitHub

main 7f51005

+15

+15

Unable to find commit in GitHub

main d3a3b36

Merge pull request #2166 from cgarciae:test-notebooks

main f647b21

Unable to find commit in GitHub

main 017d485

Unable to find commit in GitHub

main 66b4a0e

75.84% ø -0.01%

Merge pull request #2167 from cgarciae:improve-module-sphinx

main 43c8057

Unable to find commit in GitHub

main 0d12ea2

-2

-2

Unable to find commit in GitHub

main 898b159

Unable to find commit in GitHub

main ef7c5b8

Merge pull request #2188 from PWhiddy:patch-1

main a58ee9a

Merge pull request #2181 from rsokl:patch-1

main a557b13

Unable to find commit in GitHub

main 500d349

75.85% 100.00% ø

Merge pull request #2180 from cgarciae:0.5.1

main c182c7d

Unable to find commit in GitHub

main 19fc095

+6

+7

-1

75.80% 100.00% +0.01%

Add a reduction_axis argument to LayerNorm. This allows audio models to normalize across a time axis.

main 0a5a187

+2

+2

Unable to find commit in GitHub

main f095d13

Updates dynamic_scale in training state.

main 9d4505b

Fixes small typos in design_notes/module_lifecycle.rst

main b3236ce

Merge pull request #2158 from jheek:forward-unroll-arg

main 8a148db

< Previous Next >

Loading

Navigation	Overlay
`t` Navigate files	`h` Toggle hits
`y` Change url to tip of branch	`m` Toggle misses
`b / v` Jump to prev/next hit line	`p` Toggle partial
`z / x` Jump to prev/next missed or partial line	`1..9` Toggle flags
`shift + o` Open current page in GitHub	`a` Toggle all on
`/ or ?` Show keyboard shortcuts dialog	`c` Toggle context lines or commits

alt + click any link to open page in GitHub.