google/flax

Pull Requests Commits

First part of RNN refactor.

chiamp committed 2 years ago

b725d34d

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

Merge pull request #3059 from Neilblaze:patch-fix-v2

a-googler committed 2 years ago

6916f4dd

updated var & corrected typo

Neilblaze committed 2 years ago

Verified 2a1bef4f

updated to constructor

Neilblaze committed 2 years ago

Verified 18a21855

fixed broken coverage link

chiamp committed 2 years ago

39a6656e

Older