google/flax

Pull Requests Commits

fixed missing separator for rng fold in

chiamp committed 2 years ago

eb82a6cc

Bump imagenet Flax to 0.6.5

IvyZX committed 2 years ago

Verified 667a344f

Merge pull request #3114 from google:explicit-dropout-rng

a-googler committed 2 years ago

b87f5b32

allow manually specifying the rng key for Dropout

cgarciae committed 2 years ago

4bed5e92

added dict migration guide

chiamp committed 2 years ago

e48a2912

Set orbax-as-backend as default for checkpointing.

IvyZX committed 2 years ago

380fef6d

Copybara import of the project:

levskaya committed 2 years ago

63df54ad

Merge pull request #2901 from google:cgarciae-getattribute-no-recursive-register

a-googler committed 2 years ago

cc0aada0

Merge pull request #2976 from google:dependabot/pip/examples/ogbg_molpcba/tensorflow-2.11.1

a-googler committed 2 years ago

34113784

Replace gda_serialization with array_serialization as that's the new name of the library. This is a simple name replacement with no change in functionality.

yashk2810 committed 2 years ago

1145d5e0

Merge pull request #3085 from chiamp:unreferenced_docs

a-googler committed 2 years ago

1785f126

Merge pull request #3079 from chiamp:remove_advanced

a-googler committed 2 years ago

7377d654

Merge pull request #3089 from google:0.6.10

a-googler committed 2 years ago

d1385b0b

cgarciae committed 2 years ago

836aa837

Add test case for degenerate and deterministic attention layer dropout.

levskaya committed 2 years ago

fda64dc4

removed advanced_topics folder and updated link pointers

chiamp committed 2 years ago

67c96403

remove unreferenced docs

chiamp committed 2 years ago

ccfa6615

Merge pull request #3081 from chiamp:fix_rtd_urllib

a-googler committed 2 years ago

bc8fdf62

fix RTD urllib3 error

chiamp committed 2 years ago

e5922969

Merge pull request #3034 from google:use-seq-lenghts

a-googler committed 2 years ago

e9edd2ab

improve submodule binding logic

cgarciae committed 2 years ago

5e75d5b7

Merge pull request #3068 from chiamp:api_ref

a-googler committed 2 years ago

74579637

Merge pull request #3063 from chiamp:extracting_intermediates

a-googler committed 2 years ago

b7a10e86

updated api reference

chiamp committed 2 years ago

3df449ff

updated api reference

chiamp committed 2 years ago

7b3c5b4c

updated extracting_intermediates doc

chiamp committed 2 years ago

63775bee

Allow clipping the dynamic loss scale to a minimum value

a-googler committed 2 years ago

bb76e47c

Merge pull request #3058 from chiamp:readme

a-googler committed 2 years ago

12b0245d

linen_meta_test: make test compatible with JAX custom prng

Jake VanderPlas committed 2 years ago

2d47c931

Make flax tests compatible with jax.enable_custom_prng

Jake VanderPlas committed 2 years ago

fdf877cc

Newer Older