google/flax

Pull Requests Commits

Fix batch size in test case.

Marvin182 committed 5 years ago

331598b3

Merge pull request #752 from bkkaggle/master

andsteing committed 5 years ago

Verified 8b3b2949

Merge pull request #794 from tokusumi/doc-math-expr-display

avital committed 5 years ago

Verified fb04dbb4

Merge pull request #795 from tokusumi/doc-fix-breaking-italic-link

andsteing committed 5 years ago

Verified cfc714ad

Merge pull request #672 from BertrandRdp:master

a-googler committed 5 years ago

03590a4e

Link to stable Colab instance.

Marvin182 committed 5 years ago

f23cb382

Fix lint errors.

Marvin182 committed 5 years ago

9b480784

Updates PPO example flags for better consistency with other examples

andsteing committed 5 years ago

e9ccc394

Lighter - make annotated MNIST image classification example with Flax lighter

8bitmp3 committed 5 years ago

feefec92

Remove --jax_backend_target flag from main files.

Marvin182 committed 5 years ago

41f8997f

Fix missing `use_` prefix typos in documentation.

afqueiruga committed 5 years ago

Verified ae40131c

Fix breaking link in italic notation

tokusumi committed 5 years ago

ae878f36

fix math expression display

tokusumi committed 5 years ago

c1d2552c

Fold optimizer step into dropout_rng instead of splitting the dropout_rng.

Marvin182 committed 5 years ago

9fdf59f0

Marvin182 committed 5 years ago

30a5409e

Remove alternatives for packing examples.

Marvin182 committed 5 years ago

dc98e85d

Update default config.

Marvin182 committed 5 years ago

f75be406

Pass config to input_pipeline.

Marvin182 committed 5 years ago

f3ebd253

Speed up integration test.

Marvin182 committed 5 years ago

2d22dc3a

Marvin182 committed 5 years ago

a85d065c

Refactor training loop.

Marvin182 committed 5 years ago

8eee13ef

Merge pull request #778 from avital:separate-howtos

a-googler committed 5 years ago

d57e7e14

Expose Guided Tour directly on RTD left navbar

avital committed 5 years ago

7a447b1c

Reorg docs left pane, rename "patterns" to HOWTOs

avital committed 5 years ago

3c4f2413

Document `make_rng` and correct docstring

avital committed 5 years ago

61be80f6

Merge pull request #770 from avital/avital-notes

avital committed 5 years ago

Verified 012f94d5

Standardize doc names with capital letters

avital committed 5 years ago

a34f5268

Remove installation output, fix formatting

8bitmp3 committed 5 years ago

a9277c07

Use jax.nn.one_hot instead of a custom one-hot encoder

8bitmp3 committed 5 years ago

3110203b

Add PRNG design notes to help new JAX users, refactor the train_epoch explanation

8bitmp3 committed 5 years ago

88011afe

Newer Older