google/flax

Pull Requests Commits

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

Adds config.log_every_steps=100

andsteing committed 4 years ago

df7ed784

Log metrics using clu.metric_writers.

andsteing committed 4 years ago

f700254e

Fix feedback from 8bitmp3

jheek committed 4 years ago

a9c7367d

Update docs/design_notes/lift.md

jheek committed 4 years ago

5914bbdf

Apply suggestions from code review

jheek committed 4 years ago

652db91a

Fix spelling mistakes

jheek committed 4 years ago

817f7889

Add design note on lifted transformations

jheek committed 4 years ago

9de13337

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

Fix git command that add upstream

shahyash10 committed 4 years ago

d60c94c4

Merge pull request #1259 from bastings:fix_colab

a-googler committed 4 years ago

794c4e4b

Remove unused matplotlib import

Conchylicultor committed 4 years ago

d1981f9c

Create tempdir for mathplotlib, if no MPLCONFIGDIR has been specified.

tutmann committed 4 years ago

29bdb59e

Merge pull request #1262 from jheek:fix-fancy-index

a-googler committed 4 years ago

a6282dba

Fix behavior of Embed when embedding is a numpy array

jheek committed 4 years ago

a9eeb774

Merge pull request #1258 from pschuh:jnp-array

a-googler committed 4 years ago

279f80be

Newer Older