google/flax

Pull Requests Commits

Internal only change

Mohit Reddy committed 5 years ago

70734926

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

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

Update README to linen

BertrandRdp committed 5 years ago

Verified 9ede72ae

Clarify that linen docstrings aren't ready yet

avital committed 5 years ago

d7d56256

Link to Linen reference docs from RTD; specify that `flax.nn` is deprecated

avital committed 5 years ago

94feaf5e

Merge pull request #581 from google:avital-linen-out-of-alpha

a-googler committed 5 years ago

8d7c3947

Graduate Linen out of alpha

avital committed 5 years ago

Verified 9865a222

Merge pull request #560 from jheek:input-checks

a-googler committed 5 years ago

18d82dd4

remove redundant assert

jheek committed 5 years ago

a83048dd

improved Embed input type check

nestordemeure committed 5 years ago

Verified b8ad7097

add the silu activation function

TFUsers committed 5 years ago

a6f4e814

Merge pull request #508 from wrzadkow:ppo-linen

a-googler committed 5 years ago

d860ba16

Removing unused import.

freddyaboulton committed 5 years ago

837bffab

Add support for single input Conv and ConvTranspose in linen.

freddyaboulton committed 5 years ago

02d49dc2

Fix linen wmt example cache init handling bug.

levskaya committed 5 years ago

8ed5af59

Older