google/flax

Pull Requests Commits

Merge pull request #1021 from google:avital-patch-5

a-googler committed 4 years ago

9e098fdd

Link from Linen README to the reference docs

avital committed 4 years ago

Verified edcf9401

Merge pull request #1020 from jheek:fix-init-type-annotation

a-googler committed 4 years ago

1fdf7c58

Fix init/apply return annotation

jheek committed 4 years ago

e36b3f18

Generalize attribute assignment to support passing module instances

jheek committed 4 years ago

9bf71619

set reverse_translation to config.reverse_translation

a-googler committed 4 years ago

4688a754

Merge pull request #683 from jheek:use_running_average-on-__call__

a-googler committed 4 years ago

8312999b

fix writing hparams crash

a-googler committed 4 years ago

feb9119d

Fix learning rate schedule uses unscaled learning rate instead of batch size adjusted one.

jheek committed 4 years ago

5895ea68

fix comments in struct.py.

a-googler committed 4 years ago

ad18d9fc

Fix alignment of diff view for LR schedule HOWTO

avital committed 4 years ago

95a5e79a

Merge pull request #1000 from jheek:Fix-scan-broadcast

a-googler committed 4 years ago

c9a9e0b8

Fix formatting and indent.

a-googler committed 4 years ago

615f40be

Merge pull request #1002 from avital/order-of-howtos

avital committed 4 years ago

Verified 18b84518

Set order of HOWTOs explicitly on left readthedocs navbar

avital committed 4 years ago

09345bb3

Merge pull request #980 from joaogui1/howto-schedule

marcvanzee committed 4 years ago

Verified 8c6a4de7

add deprecation warning to docstrings

joaogui1 committed 4 years ago

2a22068d

Merge pull request #985 from google/howto-ensembling

marcvanzee committed 4 years ago

Verified e09b4123

Apply suggestions from code review

marcvanzee committed 4 years ago

Verified 936c3055

Remove some lines

marcvanzee committed 4 years ago

2cb72d2b

Enable broadcast in lift.scan

jheek committed 4 years ago

a374dccd

add module level warning

joaogui1 committed 4 years ago

1283b421

accpeted formating suggestions

joaogui1 committed 4 years ago

91c09154

Create module.py

marcvanzee committed 4 years ago

Verified f208f01f

Merge branch 'master' into howto-schedule

joaogui1 committed 4 years ago

ea72ea05

Merge pull request #990 from google/marcvanzee-patch-5

marcvanzee committed 4 years ago

Verified 6c6b7fc7

Merge branches 'master' and 'howto-ensembling' of https://github.com/google/flax into howto-ensembling

marcvanzee committed 4 years ago

6a41df35

Adds CSS for displaying codediff tables correctly

marcvanzee committed 4 years ago

Verified fb0995d3

joaogui1 committed 4 years ago

62f4f84c

fixed title and changed explanation

joaogui1 committed 4 years ago

10609547

Newer Older