google/flax

Pull Requests Commits

a-googler committed 3 years ago

1383ba71

Merge pull request #2389 from mattjj:skip-buggy-test

a-googler committed 3 years ago

890a3913

skip remat test that fails with autodiff

mattjj committed 3 years ago

9b06e96c

Merge pull request #2372 from cgarciae:test-docstrings

a-googler committed 3 years ago

f7545411

Merge pull request #2374 from levskaya:doctweaks2

a-googler committed 3 years ago

2f4628c9

remove whitespace

cgarciae committed 3 years ago

a04384f6

Add option `auto_flush` to `flax.metrics.tensorboard.SummaryWriter`

duckworthd committed 3 years ago

ba5ce6ce

Copies `dynamic_scale.py` from `optim/` to `training/`.

andsteing committed 3 years ago

868d3595

Allow linen's Conv layer to operate on arbitrary-rank inputs.

a-googler committed 3 years ago

e1712906

Merge pull request #2369 from Jeevesh8:patch-1

a-googler committed 3 years ago

4a0f78ac

updated supported transforms in lifting docs.

levskaya committed 3 years ago

6d561493

test docstrings with autodoc on CI

cgarciae committed 3 years ago

099642e3

Merge pull request #2373 from levskaya:doctweaks

a-googler committed 3 years ago

131fcf62

Add some docstrings to the old flax.training.common_utils module.

levskaya committed 3 years ago

0b960eb8

Update ``state_params.md`` alongside ``state_params.ipynb``

Jeevesh Juneja committed 3 years ago

Verified cae20c38

Correct state variable name

Jeevesh Juneja committed 3 years ago

Verified 9bf7283c

Merge pull request #2364 from cgarciae:fix-transpose-conv

a-googler committed 3 years ago

82daae36

add test using ConvTranspose with vmap

cgarciae committed 3 years ago

773e27f4

use jnp instead of np in ConvTranspose

cgarciae committed 3 years ago

e9420d48

Merge pull request #2363 from cgarciae:fix-colab-links

a-googler committed 3 years ago

d0e14591

Merge pull request #2340 from Davidnet:patch-1

a-googler committed 3 years ago

fb253d05

Merge branch 'main' into fix-colab-links

cgarciae committed 3 years ago

b27abaa6

fix colab links

cgarciae committed 3 years ago

04444882

Merge pull request #2360 from jheek:fix-rewound-scope-path

a-googler committed 3 years ago

acc4cfcf

Merge pull request #2361 from cgarciae:fix-main

a-googler committed 3 years ago

ede03a45

explicit flax.optim import

cgarciae committed 3 years ago

fcb63651

implicit flax.jax_utils import in full_eval.ipynb

cgarciae committed 3 years ago

9c597cf5

print sphinx errors on CI

cgarciae committed 3 years ago

40a3e868

Forward path to rewound Scope

jheek committed 3 years ago

3b7068f6

Internal change

andsteing committed 3 years ago

48dfc057

Older