google/flax

Pull Requests Commits

Merge pull request #648 from google:optimized_lstm

a-googler committed 5 years ago

692a62c5

Remove trailing whitespace.

bastings committed 5 years ago

e7a8d335

Merge remote-tracking branch 'origin/master' into optimized_lstm

bastings committed 5 years ago

24614827

Fix ConcretizationTypeError in jnp.split.

bastings committed 5 years ago

331bdc2c

Simplify _concat_dense using jnp.dot.

bastings committed 5 years ago

f5931862

Small typing fixes and doc-string improvements to flax/linen.

marcvanzee committed 5 years ago

3362ce1a

Merge pull request #717 from andsteing:imagenet_bugfix

a-googler committed 5 years ago

aa299206

Cherry-pick from #717

andsteing committed 5 years ago

3068565e

Adds --wait and $VM_READY_CMD.

andsteing committed 5 years ago

17954ef6

Adds $REPO, $BRANCH to imagenet command.

andsteing committed 5 years ago

be12beef

Merge branch 'master' into launch_gce

andsteing committed 5 years ago

65cfa5b7

Rename DummyDense and make _concat_dense more readable.

bastings committed 5 years ago

895aa1dc

Fixes validation steps.

andsteing committed 5 years ago

f3c77a30

Merge pull request #716 from google:version-0.3

a-googler committed 5 years ago

d59410b5

Apply suggestions from code review

bastings committed 5 years ago

Verified ed67ff13

Version bump to 0.3

Bertrand Rondepierre committed 5 years ago

bba8b120

Merge pull request #714 from google:avital-patch-1

a-googler committed 5 years ago

edc4034f

Fix Linen README link to core design tests

avital committed 5 years ago

Verified 5a706fbf

Fix type errors in PixelCNN example

jheek committed 5 years ago

840a99da

Reads checkpoints in parallel for files that support seeking.

Daniel Andor committed 5 years ago

afe0df10

Updates imagenet,mnist examples with result tables

andsteing committed 5 years ago

0c14d5f9

Adds launch_gce.py --connect

andsteing committed 5 years ago

b8f0fb84

Also copies logs to $GCS_WORKDIR_BASE.

andsteing committed 5 years ago

77587138

Creates /sudo_tmux.sh command in VM.

andsteing committed 5 years ago

3404984c

bastings committed 5 years ago

094e2d9d

Fix return types.

bastings committed 5 years ago

1f35eb4e

Fix initializers.

bastings committed 5 years ago

36fae9a8

Fix initializer for DummyDense.

bastings committed 5 years ago

3415c204

Merge with main branch

bastings committed 5 years ago

deda93be

Merge pull request #706 from avital:param-names

a-googler committed 5 years ago

9a0b06d2

Newer Older