google/flax

Pull Requests Commits

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

Merge pull request #586 from jheek:numpy-scalar-serialization

a-googler committed 5 years ago

12e9edad

Merge pull request #578 from nestordemeure:patch-1

a-googler committed 5 years ago

a220d200

add numpy scalar serialization

jheek committed 5 years ago

f8cd315b

Merge pull request #583 from google:BertrandRdp-readme-linen

a-googler committed 5 years ago

84674691

Merge pull request #582 from google/rtd-linen

levskaya committed 5 years ago

Verified 587bc1f9

improved Embed integer check

nestordemeure committed 5 years ago

Verified d523e883

Merge pull request #585 from levskaya:serialize_bf16_fix

a-googler committed 5 years ago

a28f7c23

Fix jax bfloat16 array handling in msgpack serialization.

levskaya committed 5 years ago

0bdbc173

Newer Older