Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 4, 2020

• Bump JAX version for tests.

  Show description for e44d1b5
  

  tomhennigan

  authored and

  copybara-github

  committed
  e44d1b5

  Copy full SHA for e44d1b5

Commits on Aug 30, 2020

• Rename 'master' to 'main' in the JAX APIs.

  Show description for b2f07aa
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  b2f07aa

  Copy full SHA for b2f07aa

Commits on Aug 28, 2020

• Implement Separable Depthwise Convolution 2D.

  Show description for db32b39
  

  Adam R. Kosiorek

  authored and

  copybara-github

  committed
  db32b39

  Copy full SHA for db32b39

Commits on Aug 26, 2020

• Only enable named call for __call__ at the moment (to support omnistaging).

  Show description for 44e0233
  

  tomhennigan

  authored and

  copybara-github

  committed
  44e0233

  Copy full SHA for 44e0233

• Change ExponentialMovingAverage.initialize(..) to take shape/dtype not array.

  Show description for 47159ce
  

  tomhennigan

  authored and

  copybara-github

  committed
  47159ce

  Copy full SHA for 47159ce

Commits on Aug 25, 2020

• Fix illegal vmap() out_axes specification in impala_lite example.

  Show description for feac073
  

  hawkinsp

  authored and

  copybara-github

  committed
  feac073

  Copy full SHA for feac073

• [hk] Update comment in impala_lite

  Show description for 16b8f5c
  

  trevorcai

  authored and

  copybara-github

  committed
  16b8f5c

  Copy full SHA for 16b8f5c

Commits on Aug 23, 2020

• Increase timeout for test.

  Show description for 1fea32b
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  1fea32b

  Copy full SHA for 1fea32b

Commits on Aug 14, 2020

• Update named_call so that it still supports Python control flow after JAX's

  Show description for 0673817
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  0673817

  Copy full SHA for 0673817

Commits on Aug 13, 2020

• Add original_shape to ParamContext.

  Show description for 39368de
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  39368de

  Copy full SHA for 39368de

Commits on Aug 12, 2020

• Test thread local stack correctly handles exceptions.

  Show description for b851ee2
  

  tomhennigan

  authored and

  copybara-github

  committed
  b851ee2

  Copy full SHA for b851ee2

Commits on Aug 9, 2020

• Merge pull request google-deepmind#64 from filangel:patch-1

  Show description for aa0b0e5
  copybara-github

  committed
  aa0b0e5

  Copy full SHA for aa0b0e5

• Ensure that parameter dtype follows input dtype.

  Show description for f9ab45e
  

  tomhennigan

  authored and

  copybara-github

  committed
  f9ab45e

  Copy full SHA for f9ab45e

Commits on Aug 8, 2020

• Make optimizer init and update consistent.

  filangelos

  authored
  5f884c7

  Copy full SHA for 5f884c7

Commits on Aug 7, 2020

• Fix bug in Haiku ConvNDTranspose, bias didn't take input dtype into account

  Show description for 8cf6247
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  8cf6247

  Copy full SHA for 8cf6247

Commits on Aug 5, 2020

• [tfds] Fix tfds.as_numpy which now returns a reusable iterable.

  Show description for 64b268e
  

  Conchylicultor

  authored and

  copybara-github

  committed
  64b268e

  Copy full SHA for 64b268e

• [tfds] Fix tfds.as_numpy which now returns a reusable iterable.

  Show description for 7203ae8
  

  Conchylicultor

  authored and

  copybara-github

  committed
  7203ae8

  Copy full SHA for 7203ae8

• [tfds] Fix tfds.as_numpy which now returns a reusable iterable.

  Show description for 329ef3c
  

  Conchylicultor

  authored and

  copybara-github

  committed
  329ef3c

  Copy full SHA for 329ef3c

Commits on Aug 3, 2020

• Replace frozendict with FlatMapping.

  Show description for 81f6f7a
  

  tomhennigan

  authored and

  copybara-github

  committed
  81f6f7a

  Copy full SHA for 81f6f7a

Commits on Aug 1, 2020

• Small tolerance bump on VQ-VAE.

  Show description for 6ae0e9c
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  6ae0e9c

  Copy full SHA for 6ae0e9c

Commits on Jul 31, 2020

• Add method interceptor API.

  Show description for 7e84ed1
  

  tomhennigan

  authored and

  copybara-github

  committed
  7e84ed1

  Copy full SHA for 7e84ed1

Commits on Jul 29, 2020

• Merge pull request google-deepmind#60 from matpalm:patch-1

  Show description for 44b4ef1
  copybara-github

  committed
  44b4ef1

  Copy full SHA for 44b4ef1

• Bump version to 0.0.2.

  Show description for 7977598
  

  LenaMartens

  authored and

  copybara-github

  committed
  7977598

  Copy full SHA for 7977598

• Use select rather than _cond.

  Show description for 8079063
  

  tomhennigan

  authored and

  copybara-github

  committed
  8079063

  Copy full SHA for 8079063

Commits on Jul 27, 2020

• Add original_dtype to ParamContext. This allows custom_getters to reason about the original call to get_parameter and coordinate better with their corresponding custom_creators.

  Show description for 95a719d
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  95a719d

  Copy full SHA for 95a719d

Commits on Jul 26, 2020

• Remove no longer necessary py3 lint config.

  Show description for a4b973b
  

  inoryy

  authored and

  copybara-github

  committed
  a4b973b

  Copy full SHA for a4b973b

Commits on Jul 24, 2020

• Update index.rst

  Show description for 4e8bb57
  matpalm

  authored
  4e8bb57

  Copy full SHA for 4e8bb57

• Fix bug in imagenet example which caused only half the test set to be used.

  Show description for 8c79047
  

  tomhennigan

  authored and

  copybara-github

  committed
  8c79047

  Copy full SHA for 8c79047

Commits on Jul 23, 2020

• Use apply_rng=True in docs.

  Show description for 70b781e
  

  tomhennigan

  authored and

  copybara-github

  committed
  70b781e

  Copy full SHA for 70b781e

Commits on Jul 21, 2020

• Port GroupNorm from Sonnet to Haiku.

  Show description for d7e0e60
  

  tomhennigan

  authored and

  copybara-github

  committed
  d7e0e60

  Copy full SHA for d7e0e60

Commits on Jul 20, 2020

• Update TODO to reference existing PEP for fully generic function protocols.

  Show description for 1510ba6
  

  aslanides

  authored and

  copybara-github

  committed
  1510ba6

  Copy full SHA for 1510ba6

Commits on Jul 16, 2020

• Fix: Use without_apply_rng in MNIST GAN example.

  Show description for 7f78e96
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  7f78e96

  Copy full SHA for 7f78e96

Commits on Jul 15, 2020

• Make pooling modules optionally batched.

  Show description for ae9fc72
  

  hamzamerzic

  authored and

  copybara-github

  committed
  ae9fc72

  Copy full SHA for ae9fc72

• Make ResetCore optionally batched.

  Show description for fe85f4f
  

  hamzamerzic

  authored and

  copybara-github

  committed
  fe85f4f

  Copy full SHA for fe85f4f

Commits on Jul 14, 2020

• Improve an error message.

  Show description for f6e9936
  

  Haiku Contributor

  authored and

  copybara-github

  committed
  f6e9936

  Copy full SHA for f6e9936

Pagination

Previous

Next