google/flax

Pull Requests Commits

a-googler committed 5 years ago

4edb8e63

Merge pull request #600 from andsteing:tests

a-googler committed 5 years ago

8d4acd07

Add offset argument to Adafactor optimizer for fine-tuning schedules.

levskaya committed 5 years ago

7a0e1df8

Merge pull request #597 from jheek:pretty-modules

a-googler committed 5 years ago

52b7f197

Use tf.io.gfile for I/O operations on benchmarks to allow remote FS access.

Mohit Reddy committed 5 years ago

4c928061

Merge pull request #608 from levskaya/linen_intro_fix

avital committed 5 years ago

Verified 06a2ad56

Update linen intro notebook to handle new mutated collection behavior of apply.

levskaya committed 5 years ago

d14762f3

Adds missing newline at end of file.

andsteing committed 5 years ago

6803d6ff

Adds two missing absltest.main()

andsteing committed 5 years ago

a1d9b357

Add a __repr__ impl to Module

jheek committed 5 years ago

e0041d06

Merge pull request #599 from avital:fix-deprecation-warning

a-googler committed 5 years ago

b8befd49

Fix incorrect reference to DeprecationWarning

avital committed 5 years ago

5aea0f9d

Bugfix for `p_pred_step` in the lm1b example.

crazydonkey200 committed 5 years ago

bb317016

Deprecate prefetch iterator

jheek committed 5 years ago

1e972542

Merge pull request #542 from BertrandRdp/master

BertrandRdp committed 5 years ago

Verified dcbd8463

Merge pull request #596 from jheek:frozen-dict-pretty-repr

a-googler committed 5 years ago

b3c0ffba

add final newline assertion and comment

jheek committed 5 years ago

484597ba

Add pretty printing for FrozenDict

jheek committed 5 years ago

874e62c0

Fix typo in docs

a-googler committed 5 years ago

397d63c4

Apply padding if only key_paddings are given. This is the general case because query paddings don't have an influence on attention anyway (the softmax is taken over keys). Also removes assumptions about self-attention when computing the other masks.

a-googler committed 5 years ago

aa3b9ae9

Merge pull request #590 from jheek:remove-non-hashable-static-args

a-googler committed 5 years ago

5e070f75

Merge pull request #550 from google:readme-discussions

a-googler committed 5 years ago

02ee7910

wrap module hash and keep Scope hash

jheek committed 5 years ago

fa1300c1

Merge pull request #563 from jpuigcerver:checkpoint-restore-target-none

a-googler committed 5 years ago

bfe304cd

jheek committed 5 years ago

81080b91

Scope should be unhashable

jheek committed 5 years ago

33600fae

rewrite jits with dict arguments

jheek committed 5 years ago

f26e03fb

add __hash__ for unbound modules

jheek committed 5 years ago

d96b6a95

Merge pull request #551 from freddyaboulton:547-kernel-size-int

a-googler committed 5 years ago

55d4a491

Modifying linen Conv/Conv Transpose to accept int for kernel size and input without a batch dim.

freddyaboulton committed 5 years ago

c5dddd40

Older