google/flax

Pull Requests Commits

Improve error message when you try to write to a read-only attribute (e.g., a read-only @property).

edloper committed 3 years ago

0be60390

Improves an error message in traverse_util.

a-googler committed 3 years ago

f40a42ed

Remove spurious call to Module.apply in test_perturb.

a-googler committed 3 years ago

8b409303

Merge pull request #2574 from 8bitmp3:remove-svn

a-googler committed 3 years ago

ab3916ac

Remove unused svn dependency in Flax CONTRIBUTING.md

8bitmp3 committed 3 years ago

Verified f0af59ca

Remove unused svn dependency in Flax setup.py

8bitmp3 committed 3 years ago

Verified bf77d243

Merge pull request #2563 from chiamp:getting_started_notebook

a-googler committed 3 years ago

bbcf96c9

Add __copy__ and __deepcopy__ methods to the _Sentinel object used to define the default value for `parent`, to simply return the sentinel object.

edloper committed 3 years ago

7d219755

Update getting_started.md

chiamp committed 3 years ago

Verified ed538915

Update getting_started.ipynb

chiamp committed 3 years ago

Verified 8e7e1908

Merge pull request #2559 from 8bitmp3:pen

a-googler committed 3 years ago

05988d1e

Merge pull request #2560 from google:lint-contributing-md

a-googler committed 3 years ago

d1bcf02f

Lint Flax Contributing guide

8bitmp3 committed 3 years ago

Verified bf88a64d

Merge branch 'IvyZX:pen' into pen

8bitmp3 committed 3 years ago

Verified 8f9acf42

Remove extra metadata in Checkpointing guide

8bitmp3 committed 3 years ago

9ca7687f

Merge pull request #2530 from IvyZX:pen

a-googler committed 3 years ago

81268fcf

Merge pull request #3 from 8bitmp3/pen

IvyZX committed 3 years ago

Verified 2616e05e

Update Flax Checkpointing guide

8bitmp3 committed 3 years ago

163364c0

Merge pull request #2548 from cgarciae:fix-autoreload

a-googler committed 3 years ago

f0e3c46d

[JAX] Change the default pmap() ordering to match the ordering of jax.devices() for single-process TPU jobs.

hawkinsp committed 3 years ago

4feaadbc

Remove pointless comprehension in jax_utils pmap helper.

levskaya committed 3 years ago

3747c877

Merge pull request #2487 from yiyixuxu:main

a-googler committed 3 years ago

5cf04bcf

Remove optional import of jax.experimental.array for older JAX versions.

IvyZX committed 3 years ago

c2268525

add default parent_ref

cgarciae committed 3 years ago

f3c255ee

Merge pull request #2553 from jheek:fix-max-

a-googler committed 3 years ago

d6877ff1

Merge pull request #2546 from google:flax-contributing-md

a-googler committed 3 years ago

064e6fc2

cap dynamic scale to float32 max

jheek committed 3 years ago

27fd2284

Merge pull request #2394 from cgarciae:transfer-learning-guide

a-googler committed 3 years ago

fad39652

set _parent_ref default value

cgarciae committed 3 years ago

5d4324b2

transfer learning guide

cgarciae committed 3 years ago

cac954e8

Older