google/flax

Pull Requests Commits

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

Add checkpointing documentation and fix clash of jupyter and tailing_whitespace

IvyZX committed 3 years ago

0cb9e99a

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

Add documentation on checkpointing

IvyZX committed 3 years ago

f9cde5d9

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

Explicitly use JAX private test utils

Jake VanderPlas committed 3 years ago

e29c9b6a

Delete flax.optim.rst

ppwwyyxx committed 3 years ago

Verified 6e4496d7

Adding link to Bayesian Inference example that uses Flax

zaxtax committed 3 years ago

08afb18c

Add Tensorstore back to required dependencies.

IvyZX committed 3 years ago

Verified 38d0dd13

Add save_checkpoint_multiprocess to api reference.

IvyZX committed 3 years ago

Verified cadb4f9e

Merge pull request #2518 from IvyZX:gcs-gfile

a-googler committed 3 years ago

b7d28402

Merge pull request #2366 from cgarciae:improve-landing-page-example

a-googler committed 3 years ago

d493d445

Newer Older