google/flax

Pull Requests Commits

Support multiple None and UNCONSTRAINED when resolving logical rules

marksandler2 committed 25 days ago

47b29f04

[flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit

danielsuo committed 28 days ago

872d50b8

[flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit.

danielsuo committed 28 days ago

881b8ed4

Avoid passing `concrete` argument to `jax.remat`

Jake VanderPlas committed 28 days ago

ec85cddf

[flax:examples:ppo] Fix some linter / import issues. #jax-fixit

danielsuo committed 30 days ago

59cfd99e

[flax:examples:sst2] Fix notebook error.

danielsuo committed 30 days ago

5a3d1489

Merge pull request #5094 from thijs-vanweezel:filters

a-googler committed 31 days ago

a5af24c2

[flax:examples:sst2] Fix pytype errors.

danielsuo committed 31 days ago

6ac5a783

[flax:examples:gemma] Fixing linter errors.

danielsuo committed 31 days ago

324f4b4b

[flax:examples:vae] Small linter fixes.

danielsuo committed 31 days ago

88d74979

[flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb.

danielsuo committed 31 days ago

9ed59e2f

[flax:examples:wmt] Small linter fixes.

danielsuo committed 31 days ago

c29b6d6b

Merge pull request #5095 from google:jax-minver-0.8.1

a-googler committed 43 days ago

697f4e5c

update jax minver to 0.8.1

cgarciae committed 44 days ago

b424c49e

hint that key is cast to string

thijs-vanweezel committed 44 days ago

a7bb109b

test pathcontains with exact=false

thijs-vanweezel committed 44 days ago

12b2e42f

allow substring matching

thijs-vanweezel committed 44 days ago

3d340558

Merge pull request #5097 from google:fix-main-4

a-googler committed 44 days ago

bee81ed4

Fixed deprecated .value usage failing CI tests

vfdev-5 committed 44 days ago

00cf44e5

Merge pull request #5084 from samanklesaria:variable_eq

a-googler committed 45 days ago

09537f49

Merge pull request #5052 from google:deprecate-value-2

a-googler committed 45 days ago

b7e66b74

deprecate .value

cgarciae committed 46 days ago

a2eac321

Merge pull request #5055 from mohsinm-dev:fix-bound-method-auto-unbinding

a-googler committed 46 days ago

d6e8e14f

Merge pull request #5073 from google:fix-nnx-scan-carry-pytree-sow

a-googler committed 46 days ago

544a0505

Merge pull request #5071 from chapman20j:set_mode_info

a-googler committed 46 days ago

0b1d841c

Merge branch 'main' into pr/5055

samanklesaria committed 46 days ago

dece6b34

Merge pull request #5090 from google:fix-main-3

a-googler committed 48 days ago

5109e2c5

cgarciae committed 48 days ago

d982784d

backfill eager_sharding in set_metadata

Cristian Garcia committed 49 days ago

1ac99b09

add HijaxVariable [rollforward]

Cristian Garcia committed 49 days ago

b5db513f

Older