google/flax

Pull Requests Commits

Add `make_array_from_single_device_arrays` to prepare to rename of the concrete `Array` to `ArrayImpl`.

yashk2810 committed 3 years ago

b71bbd66

Merge pull request #2476 from IvyZX:perturb

a-googler committed 3 years ago

cc88a73f

Merge pull request #2451 from dslisleedh:avg_pool_countpad

a-googler committed 3 years ago

b3440228

Add perturb() to module.py

IvyZX committed 3 years ago

8c0a60ae

Add test_avg_pool_no_batch under test.linen.linen_test.PoolTest

dslisleedh committed 3 years ago

d4fa16d3

Support jax.experimental.array checkpointing.

IvyZX committed 3 years ago

45c29559

Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix div_shape for it raises error when there's no batch dimension

dslisleedh committed 3 years ago

c73a4b79

Merge pull request #2440 from cgarciae:better-attribute-error-msg

a-googler committed 3 years ago

65b65ce2

Merge pull request #2465 from google:proper-scikit-learn-dependency

a-googler committed 3 years ago

0521b88c

Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix div_shape for it raises error when there's no batch dimension

dslisleedh committed 3 years ago

8af16050

Merge pull request #2425 from cgarciae:conv-docs

a-googler committed 3 years ago

1613eab9

improve attribute error msg for unbounded modules

cgarciae committed 3 years ago

c276de13

Merge pull request #2458 from cgarciae:split-pytype-ci

a-googler committed 3 years ago

fb8b640b

Merge pull request #2470 from IvyZX:gfile

a-googler committed 3 years ago

c93d7413

add proper scikit-learn depedency

cgarciae committed 3 years ago

17ed1911

Merge pull request #2473 from cgarciae:fix-state-params-docs

a-googler committed 3 years ago

cad63d1b

Fix the directory path name postfix difference when gfile.listdir reads from GCS

IvyZX committed 3 years ago

6d26f3ca

fix update_step in state_params

cgarciae committed 3 years ago

70245506

Merge branch 'IvyZX:gfile' into gfile

GallagherCommaJack committed 3 years ago

Verified 8788b6f1

Use gfile.listdir instead of gfile.glob for checkpointing. The former is proven to be much more efficient in GCS.

IvyZX committed 3 years ago

b1851f85

check for gda folders not gda files

GallagherCommaJack committed 3 years ago

Verified c5352208

Use gfile.listdir instead of gfile.glob for checkpointing. The former is proven to be much more efficient in GCS.

IvyZX committed 3 years ago

d9e6e44b

Refactor save_checkpoint out to two APIs, for single & multi process uses. And some other bug fixes.

IvyZX committed 3 years ago

6fd24460

ignore tf deprecation warning.

levskaya committed 3 years ago

3c091800

Merge pull request #2457 from cgarciae:fix-checkpoint

a-googler committed 3 years ago

59280a36

Add Array support (which is fully addressable) to msgpack serialize.

yashk2810 committed 3 years ago

b076dd9b

correct argnum indexes for transforms.checkpoint

cgarciae committed 3 years ago

f384b331

create test matrix to speedup tests

cgarciae committed 3 years ago

88517f6c

add static_argnums to nn.checkpoint

cgarciae committed 3 years ago

9522bc0b

Merge pull request #2446 from jheek:remove-pytype-generic-workaround

a-googler committed 3 years ago

e320e11c

Older