google/flax

Pull Requests Commits

Update flax/linen/transforms.py

n2cholas committed 4 years ago

de51cf3d

Fix linen.scan docstring example.

n2cholas committed 4 years ago

dc78886b

Updates MNIST from Colab.

andsteing committed 4 years ago

9ce62581

Merge pull request #1301 from google/andsteing-patch-1

jheek committed 4 years ago

Verified 63bd1339

Updates examples/mnist to use Optax.

andsteing committed 4 years ago

c4b34644

Ignore `jax.host_{id,count}()` deprecation warning

andsteing committed 4 years ago

Verified 2a459324

Merge pull request #1300 from google/andsteing-patch-1

jheek committed 4 years ago

Verified 87a483b2

Bumps JAX version to 0.2.13

andsteing committed 4 years ago

Verified a9e3f3cb

Merge pull request #1280 from google/marcvanzee-patch-10

marcvanzee committed 4 years ago

Verified d804b906

Merge pull request #1274 from jheek/update-install-instructions

marcvanzee committed 4 years ago

Verified ac75341b

Merge pull request #1281 from yashs97/fix-contributing-doc

marcvanzee committed 4 years ago

Verified 4f5438e0

Shows boilerplate cell by default to make sure it's executed.

andsteing committed 4 years ago

c6877e94

[flex/linen]: Make `rngs` type "Optional" to avoid pytype errors.

liviosoares committed 4 years ago

0848d9cf

Merge pull request #1284 from jheek:init-mutable

a-googler committed 4 years ago

2faf0eaf

Expose `dot_product_attention_weights`

avital committed 4 years ago

37d10fe4

fix recursion bug

jheek committed 4 years ago

de2737e3

add test for mutable=True

jheek committed 4 years ago

8f3c78a6

Fix intermediates during init:

jheek committed 4 years ago

ab7db815

Remove git command that fetches all branches from upstream

shahyash10 committed 4 years ago

2aa3430f

Merge pull request #1275 from jheek:cached-lifted-jit

a-googler committed 4 years ago

61280ffb

add nonlocal to jitted function for readability

jheek committed 4 years ago

c8ecb95f

Renames `update_step()` to `train_step()`.

andsteing committed 4 years ago

e44115c6

Merge pull request #1278 from jakevdp/typos

avital committed 4 years ago

Verified 38782d09

Fix minor typos

shahyash10 committed 4 years ago

c47ea262

Update CHANGELOG.md

marcvanzee committed 4 years ago

Verified 3b80a646

Fix some typos in contributing.md

jakevdp committed 4 years ago

66309bf9

add type annotations

jheek committed 4 years ago

2b7582a5

Add proper cache behavior for lift.jit

jheek committed 4 years ago

9c19c13b

Update install instructions for contributing.

jheek committed 4 years ago

e2e4b909

Merge pull request #1272 from yashs97/fix-adding-upstream

marcvanzee committed 4 years ago

Verified 9696c739

Newer Older