google/flax

Pull Requests Commits

this should be a no-op

avital committed 3 years ago

7962373a

Internal change

marcvanzee committed 3 years ago

ef293be8

Cleans up some code

marcvanzee committed 3 years ago

bd84383c

Add vmap_with_axes, a logical axis version of nn.vmap

James Lee-Thorp committed 3 years ago

fb771186

Merge pull request #1825 from marcvanzee:pipy

a-googler committed 3 years ago

78d0566c

Merge branch 'main' of https://www.github.com/google/flax into pipy

marcvanzee committed 3 years ago

695a8805

Bump Flax to v0.4.0

marcvanzee committed 3 years ago

5de87faf

Merge pull request #1824 from marcvanzee:rtd-deprecated

a-googler committed 3 years ago

a6e77441

Updates Linen README

marcvanzee committed 3 years ago

a7cf071d

Updates core notebook to Linen

marcvanzee committed 3 years ago

1c4fe91d

Removes deprecated API from docs

marcvanzee committed 3 years ago

12a1159c

Actually removes the deprecated folder.

marcvanzee committed 3 years ago

aca40ad7

Removes flax.deprecated.nn.

marcvanzee committed 3 years ago

948ee260

Merge pull request #1815 from NeilGirdhar:container

a-googler committed 3 years ago

7288ada4

Merge pull request #1804 from jheek:non-float32-normalization

a-googler committed 3 years ago

ca683887

Add support for non-float32 normalization for linen normalization layers.

jheek committed 3 years ago

c87b54b7

Removes PixelCNN++ example. This example continues being flakey on TPU and we never really got it working well. I checked with Jamie (the original author), and he was okay with removing it, he mentioned he may put it in a personal repo at some point in the future.

marcvanzee committed 3 years ago

866ce07b

Adds prominent `scenic` link to `examples/README.md`

andsteing committed 3 years ago

6690f0c0

Merge pull request #1806 from andsteing/optax

andsteing committed 3 years ago

Verified e1cd0d34

Updates with Marc's comments.

andsteing committed 3 years ago

0de76a32

Updates the ensembling HOWTO to Optax.

andsteing committed 3 years ago

3b3c8c04

Fix missing backtick before `'CIRCULAR'` in `Conv` docstring.

dhgarrette committed 3 years ago

66a4f1d8

make params_with_axes() work when params_axes is not mutable

a-googler committed 3 years ago

2c2a7961

Make Filter a Collection instead of a Container

NeilGirdhar committed 3 years ago

a21204f0

Replaces `print()` with `logging.info()`.

andsteing committed 3 years ago

eb7e7ae7

Remove potentially confusing comment question.

melissatan committed 3 years ago

Verified 601783d3

Mentions `matthias-wright/flaxmodel` models by name for better searchability.

andsteing committed 3 years ago

addfe9c2

Merge pull request #1795 from jheek:fix-type-annotation

a-googler committed 3 years ago

04579e3a

Fix wrong type annotation.

jheek committed 3 years ago

44b6e461

Merge pull request #1723 from jheek:lazy-rng

a-googler committed 3 years ago

b60f7f45

Older