google/flax

Pull Requests Commits

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

Add OptimizedLSTMCell Linen version.

bastings committed 5 years ago

5135d952

Relax parameter attribute check

avital committed 5 years ago

ec995477

Merge pull request #680 from jheek:lift-md

a-googler committed 5 years ago

ffaf51e7

Move notes into docs

jheek committed 5 years ago

b7cd0432

Also rsync checkpoints, no final upload.

andsteing committed 5 years ago

d63bb79d

Merge pull request #647 from qiuminxu:master

a-googler committed 5 years ago

06a79634

Changes --name to config name for imagenet.

andsteing committed 5 years ago

54bf3f66

Specifies full version for clu requirement.

andsteing committed 5 years ago

00bb8c01

Merge pull request #694 from andsteing:launch_gce

a-googler committed 5 years ago

96c3cbd7

Update setup.py

qiuminxu committed 5 years ago

e0365c08

Bugfix startup_script.sh without ACCELERATOR.

andsteing committed 5 years ago

814051e4

Bumps jaxlib to 0.1.57 & uses VM image with cu110.

andsteing committed 5 years ago

c542c272

Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-

andsteing committed 5 years ago

36ca83ac

Adds clu to imagenet/requirements.txt

andsteing committed 5 years ago

e8e2a986

Update README.md

avital committed 5 years ago

Verified 7ef53e6e

Merge pull request #681 from 8bitmp3:patch-1

a-googler committed 5 years ago

b275ff5e

Merge pull request #657 from avital:docs2

a-googler committed 5 years ago

567dcfcb

Replace flax named_call with upstream version jax.named_call

qiuminxu committed 5 years ago

bcf4eefb

Update flax/nn/base.py

8bitmp3 committed 5 years ago

Verified e875ef8e

Newer Older