google/flax

Pull Requests Commits

Update gemma sampler to handle additional params

a-googler committed 309 days ago

d05075e1

[nnx] Pass kwargs to super `__init_subclass__`.

a-googler committed 309 days ago

323107d2

[nnx] copy _var_metadata

Cristian Garcia committed 309 days ago

eca312f1

Merge pull request #4542 from google:nnx-compact-2

a-googler committed 311 days ago

7e408079

[nnx] bridge Module

cgarciae committed 312 days ago

f2892823

[nnx] add cached_partial

Cristian Garcia committed 314 days ago

61e5df49

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

a-googler committed 314 days ago

b5f478ff

[nnx] fix Variable hooks

Cristian Garcia committed 315 days ago

2b870fa7

[nnx] register_variable_name refactor

Cristian Garcia committed 316 days ago

c67eb094

added support to the accuracy metric for binary classification

mattbahr committed 316 days ago

08e205f8

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

a-googler committed 316 days ago

33bf6b6a

update pypi publish

cgarciae committed 317 days ago

0fb09bf6

Merge pull request #4525 from google:v0.10.3

a-googler committed 317 days ago

dc01737a

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

a-googler committed 317 days ago

d935e53b

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

a-googler committed 320 days ago

7efdfc82

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

a-googler committed 323 days ago

429033b3

cgarciae committed 324 days ago

4c1f760a

batch_norm.rst: == should be =

cool-RR committed 325 days ago

Verified cc4279c1

Merge pull request #4515 from IvyZX:perturb

a-googler committed 326 days ago

f71ce6c6

Add Module.perturb

IvyZX committed 326 days ago

4aba3404

make gemma FFW LoRA friendly

a-googler committed 327 days ago

7f1567c1

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

a-googler committed 329 days ago

881685c7

fix LoRA initialization error in nnx layer

a-googler committed 330 days ago

00ee6f11

Merge pull request #4503 from IvyZX:varmap

a-googler committed 332 days ago

a8a192ff

Remove all Param(None) lines

IvyZX committed 333 days ago

ad642ef6

Rename variable string mapping utils and move them to variableslib

IvyZX committed 333 days ago

61634e47

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

IvyZX committed 333 days ago

a7b0edeb

[nnx] add tabulate

cgarciae committed 334 days ago

caed9ca2

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

a-googler committed 335 days ago

d28f03fe

[nnx] fix fiddle

cgarciae committed 335 days ago

13013621

Older