google/flax

Pull Requests Commits

Implement metadata flip

jheek committed 3 years ago

f4c4670c

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

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

transfer learning guide

cgarciae committed 3 years ago

cac954e8

Update Flax Contributing.md

8bitmp3 committed 3 years ago

Verified b2e714c9

Merge pull request #2510 from dslisleedh:fix_dropout

a-googler committed 3 years ago

6b80cbb2

Merge branch 'main' into fix_dropout

chiamp committed 3 years ago

Verified 98453e51

Merge pull request #2545 from IvyZX:jupytext-pin

a-googler committed 3 years ago

c6289104

Pin jupytext version in requirements.txt

IvyZX committed 3 years ago

Verified 22afba5f

Merge pull request #2544 from google:update-readme-guides-link

a-googler committed 3 years ago

3cf51791

Update Guides link to Flax README

8bitmp3 committed 3 years ago

Verified a3d8f032

Merge pull request #2507 from cgarciae:cancel-workflow-on-failure

a-googler committed 3 years ago

521f5163

Merge pull request #2540 from cgarciae:parametrize-dropout-rng-collection

a-googler committed 3 years ago

d6fb0129

Merge pull request #2486 from yotarok:fix_batch_norm_init

a-googler committed 3 years ago

376804f3

Fix flax.linen.stochastic.Dropout

dslisleedh committed 3 years ago

af129499

Added IncorrectPostInitOverrideError to capture incorrect post init overrides.

chiamp committed 3 years ago

354d29b4

add rng_collection argument to Dropout

cgarciae committed 3 years ago

d7b3c12c

Remove uses of JAX-internal test utilites from Flax.

hawkinsp committed 3 years ago

e0de630d

Merge pull request #2521 from zaxtax:add_bayesian_example

a-googler committed 3 years ago

16a95f02

Merge pull request #2529 from chiamp:init_apply_checkinstance

a-googler committed 3 years ago

315c297e

Removed trailing whitespaces in flax/linen/module.py

chiamp committed 3 years ago

919cdfb5

Merge pull request #2522 from ppwwyyxx:patch-1

a-googler committed 3 years ago

a479e13b

Merge pull request #2520 from IvyZX:add-ts

a-googler committed 3 years ago

ac56d590

zaxtax committed 3 years ago

bab8e749

Merge pull request #2519 from IvyZX:doc-fix

a-googler committed 3 years ago

5470cd37

Older