google/flax

Pull Requests Commits

Remove references to deprecated jax.ShapedArray

Jake VanderPlas committed 2 years ago

57a7f26e

Updates examples link.

andsteing committed 2 years ago

5bfa098d

[Trivial] Uptick the version after new release.

IvyZX committed 2 years ago

e17d1cec

Merge pull request #2987 from IvyZX:push

a-googler committed 2 years ago

e061c692

IvyZX committed 2 years ago

b8c89e8c

Merge pull request #2958 from chiamp:pop_doc_str

a-googler committed 2 years ago

8957389d

Merge pull request #2975 from chiamp:frozen_dict_api_doc

a-googler committed 2 years ago

396edea1

Merge pull request #2926 from google:missing-references

a-googler committed 2 years ago

dd8040e5

Copybara import of the project:

dependabot[bot] committed 2 years ago

2248f44b

Copybara import of the project:

dependabot[bot] committed 2 years ago

02e2ad13

Merge pull request #2977 from google:dependabot/pip/examples/lm1b/tensorflow-2.11.1

a-googler committed 2 years ago

f037751c

Merge pull request #2979 from google:dependabot/pip/examples/nlp_seq/tensorflow-2.11.1

a-googler committed 2 years ago

b0a78a08

Merge pull request #2980 from google:dependabot/pip/examples/sst2/tensorflow-2.11.1

a-googler committed 2 years ago

a7b0242d

Merge pull request #2983 from google:dependabot/pip/examples/imagenet/tensorflow-2.11.1

a-googler committed 2 years ago

ff55c0db

Merge pull request #2984 from google:dependabot/pip/examples/ppo/tensorflow-2.11.1

a-googler committed 2 years ago

9b13d0f4

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/ppo

dependabot[bot] committed 2 years ago

Verified b7c54860

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/imagenet

dependabot[bot] committed 2 years ago

Verified 8af934db

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/nlp_seq

dependabot[bot] committed 2 years ago

Verified 81b432e4

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/sst2

dependabot[bot] committed 2 years ago

Verified 742fe760

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/lm1b

dependabot[bot] committed 2 years ago

Verified f939c4d9

Fixed pop utility fn doc string

chiamp committed 2 years ago

7b1b4d93

Added utility functions to frozen_dict api

chiamp committed 2 years ago

e1cd5a10

Do not print fields defined with dataclasses.field(repr=False)

Conchylicultor committed 2 years ago

d384135b

Make Flax checkpointing use `sharding` instead of extracting mesh and pspecs. Also improved some warning messages.

IvyZX committed 2 years ago

41c9304c

Fixes kwargs in convert_to_graphs_tuple_fn.

andsteing committed 2 years ago

c2da6f90

Merge pull request #2770 from cgarciae:add-bidirectional

a-googler committed 2 years ago

a4a89ae4

add Bidirectional module

cgarciae committed 2 years ago

6412c6c0

Merge pull request #2967 from yechengxi:patch-2

a-googler committed 2 years ago

23340bbb

set axis='batch'

yechengxi committed 2 years ago

Verified 1c6cb6b0

Roll back the Orbax flag flip.

IvyZX committed 2 years ago

751f26a1

Newer Older