Commits

Breadcrumbs

History for

onmaster

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 6, 2022

• Delete AQT code in google_research/aqt.

  Show description for fef3211
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  fef3211

  Copy full SHA for fef3211

Commits on Mar 31, 2022

• Enable AQT quantization for MultiQueryDotAttention in Flaxformer.

  Show description for e3a8437
  

  shivaniag

  authored and

  copybara-github

  committed
  e3a8437

  Copy full SHA for e3a8437

Commits on Mar 21, 2022

• Silence some pytype errors.

  Show description for aaaef75
  

  rchen152

  authored and

  copybara-github

  committed
  aaaef75

  Copy full SHA for aaaef75

Commits on Mar 14, 2022

• Merge of PR google-research#1022

  Show description for 3dcd301
  

  ychzhang

  authored and

  copybara-github

  committed
  3dcd301

  Copy full SHA for 3dcd301

• Gcloud reproduction instructions for AQT experiments (google-research#1022)

  Show description for 88ce4b9
  ychzhang

  authored
  88ce4b9

  Copy full SHA for 88ce4b9

• INTERNAL CHANGE.

  Show description for 97d254b
  

  shivaniag

  authored and

  copybara-github

  committed
  97d254b

  Copy full SHA for 97d254b

Commits on Mar 10, 2022

• Internal changes.

  Show description for 6b7f2fb
  

  shivaniag

  authored and

  copybara-github

  committed
  6b7f2fb

  Copy full SHA for 6b7f2fb

Commits on Mar 9, 2022

• Import DynamicScale directly.

  Show description for 33fdafd
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  33fdafd

  Copy full SHA for 33fdafd

Commits on Mar 4, 2022

• Fix internal blocks.

  Show description for a300d0a
  

  shivaniag

  authored and

  copybara-github

  committed
  a300d0a

  Copy full SHA for a300d0a

Commits on Mar 3, 2022

• Update the references of flax.deprecated.nn to flax.linen and fix the commands in README.

  Show description for 2e47f2f
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  2e47f2f

  Copy full SHA for 2e47f2f

• Fix internal block in google_research.

  Show description for 767d318
  

  shivaniag

  authored and

  copybara-github

  committed
  767d318

  Copy full SHA for 767d318

Commits on Mar 1, 2022

• Remove references to internal jax.numpy._constant_like function

  Show description for f497e16
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  f497e16

  Copy full SHA for f497e16

• Changed configuration for internal changes. This should not affect existing behavior.

  Show description for ac0ca4c
  

  shivaniag

  authored and

  copybara-github

  committed
  ac0ca4c

  Copy full SHA for ac0ca4c

Commits on Feb 14, 2022

• Removes trailing whitespace.

  Show description for 8069d97
  

  shivaniag

  authored and

  copybara-github

  committed
  8069d97

  Copy full SHA for 8069d97

Commits on Feb 2, 2022

• Adding instructions to launch training for (optionally quantized) WMT model.

  Show description for f258e73
  

  shivaniag

  authored and

  copybara-github

  committed
  f258e73

  Copy full SHA for f258e73

Commits on Feb 1, 2022

• Adds fake-quant style weights only quantization config for experiments.

  Show description for fca6f76
  

  shivaniag

  authored and

  copybara-github

  committed
  fca6f76

  Copy full SHA for fca6f76

• Internal change

  Show description for 675ed8e
  

  marcvanzee

  authored and

  copybara-github

  committed
  675ed8e

  Copy full SHA for 675ed8e

Commits on Jan 28, 2022

• Replace deprecated references to jax.nn.initializers so as to prevent flaxformer OSS from working with flax==0.4.0.

  Show description for d651f13
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  d651f13

  Copy full SHA for d651f13

Commits on Jan 26, 2022

• Fix requirements for AQT library and AQT models.

  Show description for 1eec83d
  

  shivaniag

  authored and

  copybara-github

  committed
  1eec83d

  Copy full SHA for 1eec83d

Commits on Jan 20, 2022

• Make DenseAqt to have logical axis names.

  Show description for 2225378
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  2225378

  Copy full SHA for 2225378

Commits on Jan 18, 2022

• fix test

  Show description for ccc94ce
  andrewluchen

  committed
  ccc94ce

  Copy full SHA for ccc94ce

Commits on Jan 6, 2022

• Adds wmt_mlperf transformer model and input pipeline for wmt dataset. Also adds hparam configurations to optionally quantize the model.

  Show description for fb03c4d
  

  shivaniag

  authored and

  copybara-github

  committed
  fb03c4d

  Copy full SHA for fb03c4d

Commits on Dec 4, 2021

• Adds requirements file for AQT imagenet.

  Show description for 7cdec61
  

  shivaniag

  authored and

  copybara-github

  committed
  7cdec61

  Copy full SHA for 7cdec61

Commits on Dec 3, 2021

• Refactor ResNet-50 model code.

  Show description for 58eea29
  

  ychzhang

  authored and

  copybara-github

  committed
  58eea29

  Copy full SHA for 58eea29

Commits on Nov 25, 2021

• Update JAX-internal monkey patch to point to new location.

  Show description for 61f26ef
  

  hawkinsp

  authored and

  copybara-github

  committed
  61f26ef

  Copy full SHA for 61f26ef

Commits on Nov 24, 2021

• Fix setting same parameter twice in class.

  Show description for 2931045
  

  shivaniag

  authored and

  copybara-github

  committed
  2931045

  Copy full SHA for 2931045

Commits on Oct 28, 2021

• Internal change

  Show description for f72f544
  

  marcvanzee

  authored and

  copybara-github

  committed
  f72f544

  Copy full SHA for f72f544

Commits on Oct 19, 2021

• [JAX] Port lax translation rules to updated XLA translation rule API.

  Show description for 5a4d95b
  

  hawkinsp

  authored and

  copybara-github

  committed
  5a4d95b

  Copy full SHA for 5a4d95b

Commits on Oct 12, 2021

• Add link to paper

  Show description for 4ab6c16
  

  rybakov

  authored and

  copybara-github

  committed
  4ab6c16

  Copy full SHA for 4ab6c16

Commits on Oct 4, 2021

• [JAX] Increase numerical tolerances of tests in preparation for an XLA:CPU vectorization change.

  Show description for f99cb6c
  

  hawkinsp

  authored and

  copybara-github

  committed
  f99cb6c

  Copy full SHA for f99cb6c

Commits on Oct 1, 2021

• [JAX] Replace uses of deprecated jax.ops.index_update(x, idx, y) APIs with their up-to-date, more succinct equivalent x.at[idx].set(y).

  Show description for 30e5a4a
  

  hawkinsp

  authored and

  copybara-github

  committed
  30e5a4a

  Copy full SHA for 30e5a4a

Commits on Sep 23, 2021

• No functional changes. Group weight and act quantization test into their classes.

  Show description for 6167992
  

  shivaniag

  authored and

  copybara-github

  committed
  6167992

  Copy full SHA for 6167992

Commits on Sep 14, 2021

• [JAX] Replace use of jax.lax.partial with functools.partial.

  Show description for 05aaa5d
  

  hawkinsp

  authored and

  copybara-github

  committed
  05aaa5d

  Copy full SHA for 05aaa5d

Commits on Sep 7, 2021

• Create a new util function that computes precision for floating-point quantization.

  Show description for efc2156
  

  jihwanlee-alphago

  authored and

  copybara-github

  committed
  efc2156

  Copy full SHA for efc2156

Commits on Aug 31, 2021

• Ensure values passed to jax.numpy functions are arrays rather than lists.

  Show description for 45eca11
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  45eca11

  Copy full SHA for 45eca11

Pagination

Previous

Next