google/flax

Pull Requests Commits

port nlp_seq to linen

bohnetbd committed 5 years ago

aad1cc1e

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

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

Updating docstring for Conv explaining that int can be passed in.

freddyaboulton committed 5 years ago

a4ac2d0c

Mapping int to (int,) in Conv and ConvTranspose for kernel_size param.

freddyaboulton committed 5 years ago

188b1346

Removing extraneous comma

freddyaboulton committed 5 years ago

b4696cfc

kernel_size can be an int for Conv and ConvTranspose.

freddyaboulton committed 5 years ago

e9dd06c3

Merge pull request #564 from freddyaboulton:linen-single-input-conv

a-googler committed 5 years ago

7cb7c33e

Merge pull request #577 from TFUsers:master

a-googler committed 5 years ago

bd50343b

Final update including linen patterns.

Bertrand Rondepierre committed 5 years ago

7fca96c5

Older