google/flax

Pull Requests Commits

[nnx] add cache_args

cgarciae committed 1 year ago

19f78a6e

Merge pull request #4536 from mattbahr:accuracy-metric-binary-classification-support

a-googler committed 1 year ago

b5f478ff

[nnx] fix Variable hooks

Cristian Garcia committed 1 year ago

2b870fa7

[nnx] register_variable_name refactor

Cristian Garcia committed 1 year ago

c67eb094

added support to the accuracy metric for binary classification

mattbahr committed 1 year ago

08e205f8

Merge pull request #4538 from google:improve-publish-safety

a-googler committed 1 year ago

33bf6b6a

update pypi publish

cgarciae committed 1 year ago

0fb09bf6

Merge pull request #4525 from google:v0.10.3

a-googler committed 1 year ago

dc01737a

Merge pull request #4524 from cool-RR:patch-5

a-googler committed 1 year ago

d935e53b

Merge pull request #4493 from google:nnx-tabulate-2

a-googler committed 1 year ago

7efdfc82

The CL introduces a new SowConfig which is used to configure which intermediate values are to be sowed.

a-googler committed 1 year ago

429033b3

cgarciae committed 1 year ago

4c1f760a

batch_norm.rst: == should be =

cool-RR committed 1 year ago

Verified cc4279c1

Merge pull request #4515 from IvyZX:perturb

a-googler committed 1 year ago

f71ce6c6

Add Module.perturb

IvyZX committed 1 year ago

4aba3404

make gemma FFW LoRA friendly

a-googler committed 1 year ago

7f1567c1

Merge pull request #4504 from IvyZX:no-none-var

a-googler committed 1 year ago

881685c7

fix LoRA initialization error in nnx layer

a-googler committed 1 year ago

00ee6f11

Merge pull request #4503 from IvyZX:varmap

a-googler committed 1 year ago

a8a192ff

Remove all Param(None) lines

IvyZX committed 1 year ago

ad642ef6

Rename variable string mapping utils and move them to variableslib

IvyZX committed 1 year ago

61634e47

Don't create param in normalization layers instead of create None-value params.

IvyZX committed 1 year ago

a7b0edeb

[nnx] add tabulate

cgarciae committed 1 year ago

caed9ca2

Merge pull request #4500 from google:nnx-fix-fiddle

a-googler committed 1 year ago

d28f03fe

[nnx] fix fiddle

cgarciae committed 1 year ago

13013621

Update einsum layer for Gemma example

a-googler committed 1 year ago

0eeab5aa

Merge pull request #4499 from google:nnx-improve-module-docs

a-googler committed 1 year ago

e3bcc442

[nnx] improve Module docs

cgarciae committed 1 year ago

bdcc33a5

Remove usages of orbax_utils.save_args_from_target, as this function does nothing (it used to control a checkpointing behavior that has since been optimized away).

cpgaffney1 committed 1 year ago

b5d4ed85

Merge pull request #4486 from google:fix-to-nnx-linen-attributes

a-googler committed 1 year ago

e4418e27

Older