google/flax

Pull Requests Commits

VAE with a discrete bottleneck with quantization done using the kmeans algorithm.

henrykmichalewski committed 4 years ago

6a355b22

Fold optimizer step into dropout_rng instead of splitting the dropout_rng.

Marvin182 committed 4 years ago

9fdf59f0

Marvin182 committed 4 years ago

30a5409e

Remove alternatives for packing examples.

Marvin182 committed 4 years ago

dc98e85d

Update default config.

Marvin182 committed 4 years ago

f75be406

Pass config to input_pipeline.

Marvin182 committed 4 years ago

f3ebd253

Speed up integration test.

Marvin182 committed 4 years ago

2d22dc3a

Marvin182 committed 4 years ago

a85d065c

Refactor training loop.

Marvin182 committed 4 years ago

8eee13ef

Merge pull request #778 from avital:separate-howtos

a-googler committed 4 years ago

d57e7e14

Expose Guided Tour directly on RTD left navbar

avital committed 4 years ago

7a447b1c

Reorg docs left pane, rename "patterns" to HOWTOs

avital committed 4 years ago

3c4f2413

Merge pull request #770 from avital/avital-notes

avital committed 4 years ago

Verified 012f94d5

Standardize doc names with capital letters

avital committed 4 years ago

a34f5268

Switch to clu MetricWriters.

Marvin182 committed 5 years ago

b0496547

Merge pull request #757 from jheek:fix-variable-mut-check

a-googler committed 5 years ago

5d6a6796

Internal change.

Marvin182 committed 5 years ago

f9901252

Update CHANGELOG

jheek committed 5 years ago

e35361a7

Fix variable exists check

jheek committed 5 years ago

d971db56

Merge pull request #751 from andsteing/install_deps_oo

andsteing committed 5 years ago

Verified 6e6c89b2

Merge pull request #740 from jheek:single-example-pooling

a-googler committed 5 years ago

72cbc94f

Merge pull request #741 from jheek:avoid-llvm-optimization-for-long-tests

a-googler committed 5 years ago

82986b7d

Removes -v from pip install in build.yml

andsteing committed 5 years ago

c61423c9

Switches back to tf[-text]>=2.4.0

andsteing committed 5 years ago

aad92afa

Reverts 9c28c5 trying to fix build.

andsteing committed 5 years ago

8aa5c12d

disable LLVM optimization on tests with long compilation times

jheek committed 5 years ago

f948246f

Merge pull request #737 from andsteing:tests

a-googler committed 5 years ago

483c53ea

Add support for single example pooling

jheek committed 5 years ago

549b0592

Fixes some linting errors.

andsteing committed 5 years ago

a1a3efad

Makes tf[-text] version requirements >=.

andsteing committed 5 years ago

9c28c518

Older