google/flax

Pull Requests Commits

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

a-googler committed 352 days ago

33bf6b6a

update pypi publish

cgarciae committed 352 days ago

0fb09bf6

Merge pull request #4525 from google:v0.10.3

a-googler committed 352 days ago

dc01737a

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

a-googler committed 352 days ago

d935e53b

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

a-googler committed 355 days ago

7efdfc82

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

a-googler committed 358 days ago

429033b3

cgarciae committed 359 days ago

4c1f760a

batch_norm.rst: == should be =

cool-RR committed 360 days ago

Verified cc4279c1

Merge pull request #4515 from IvyZX:perturb

a-googler committed 362 days ago

f71ce6c6

Add Module.perturb

IvyZX committed 362 days ago

4aba3404

make gemma FFW LoRA friendly

a-googler committed 363 days 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

[nnx] fix ToNNX linen_attributes update

cgarciae committed 1 year ago

342a5e83

Merge pull request #4485 from google:nnx-multimetric

a-googler committed 1 year ago

969f3cb2

[nnx] fix MultiMetric typing

cgarciae committed 1 year ago

ca4dc8ca

Merge pull request #4479 from jakevdp:fix-workflows

a-googler committed 1 year ago

1961c125

Merge pull request #4478 from jakevdp:nightly-build

a-googler committed 1 year ago

6abf8bcb

Newer Older