Commits

Branch selector

 master

User selector

All users

All time

Commit History

Commits on May 20, 2024

• Replace deprecated `jax.tree_*` functions with `jax.tree.*`

  Show description for bea6d6b
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  bea6d6b

  Copy full SHA for bea6d6b

Commits on May 3, 2024

• Acme: update deprecated tree_map API

  Show description for 8d4cc04
  

  Changwan Ryu

  authored and

  copybara-github

  committed
  8d4cc04

  Copy full SHA for 8d4cc04

Commits on Apr 26, 2024

• Acme: Increase thread pool size to prevent a hang between actors and inference servers

  Show description for 3a1420d
  

  Changwan Ryu

  authored and

  copybara-github

  committed
  3a1420d

  Copy full SHA for 3a1420d

Commits on Apr 15, 2024

• Depend on the jax substrate of tensorflow_probability explicitly.

  Show description for aa42e1c
  

  ThomasColthurst

  authored and

  copybara-github

  committed
  aa42e1c

  Copy full SHA for aa42e1c

Commits on Mar 25, 2024

• Migrate users of tfp.experimental.substrates.jax to import it as

  Show description for 440df89
  

  ThomasColthurst

  authored and

  copybara-github

  committed
  440df89

  Copy full SHA for 440df89

Commits on Mar 11, 2024

• Acme: Allow use of multiple TPU accelerators for inference servers

  Show description for a04759f
  

  Changwan Ryu

  authored and

  copybara-github

  committed
  a04759f

  Copy full SHA for a04759f

Commits on Jan 3, 2024

• Added pytype None checks to savers.py

  Show description for 148331f
  

  Acme Contributor

  authored and

  copybara-github

  committed
  148331f

  Copy full SHA for 148331f

• Added pytype None checks to inference_server.py

  Show description for 4c6351e
  

  Acme Contributor

  authored and

  copybara-github

  committed
  4c6351e

  Copy full SHA for 4c6351e

Commits on Nov 16, 2023

• [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

  Show description for 1177501
  

  hawkinsp

  authored and

  copybara-github

  committed
  1177501

  Copy full SHA for 1177501

Commits on Oct 20, 2023

• Updated jax.config import

  Show description for 53eeda3
  

  superbobry

  authored and

  copybara-github

  committed
  53eeda3

  Copy full SHA for 53eeda3

Commits on Oct 17, 2023

• [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

  Show description for 43f5b68
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  43f5b68

  Copy full SHA for 43f5b68

Commits on Oct 9, 2023

• Acme: Make D4PG use the tested n-step transition adder.

  Show description for d92e23b
  

  bshahr

  authored and

  copybara-github

  committed
  d92e23b

  Copy full SHA for d92e23b

Commits on Sep 26, 2023

• [LSC] Ignore incorrect type annotations related to jax.numpy APIs

  Show description for ac668d5
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  ac668d5

  Copy full SHA for ac668d5

• Add assertion to aid type checking.

  Show description for 46c57d9
  

  Acme Contributor

  authored and

  copybara-github

  committed
  46c57d9

  Copy full SHA for 46c57d9

Commits on Sep 25, 2023

• Add assertion to aid type checking.

  Show description for 32009e2
  

  Acme Contributor

  authored and

  copybara-github

  committed
  32009e2

  Copy full SHA for 32009e2

• "Resolve unsoundness caught by pytype --strict-none-binding.

  Show description for 74a841a
  

  Acme Contributor

  authored and

  copybara-github

  committed
  74a841a

  Copy full SHA for 74a841a

Commits on Aug 30, 2023

• Use default_factory instead of an instance for this dataclass field's default value.

  Show description for 993826a
  

  yilei

  authored and

  copybara-github

  committed
  993826a

  Copy full SHA for 993826a

Commits on Aug 28, 2023

• Previously, the learning_rate parameter was only used for logging and not actually applied to constructing the default optimizer (in case one wasn't passed as an argument), which used hardcoded def…

  Show description for 44d28ae
  

  siggiorn

  authored and

  copybara-github

  committed
  44d28ae

  Copy full SHA for 44d28ae

Commits on Aug 21, 2023

• acme: pin an older version of jax to fix testing issues; update flaky test.

  Show description for 7560b96
  

  mwhoffman

  authored and

  copybara-github

  committed
  7560b96

  Copy full SHA for 7560b96

Commits on Aug 17, 2023

• [JAX] Fix incorrect type annotations.

  Show description for c7690d1
  

  hawkinsp

  authored and

  copybara-github

  committed
  c7690d1

  Copy full SHA for c7690d1

Commits on Aug 10, 2023

• [JAX] Replace uses of `jnp.array` in types with `jnp.ndarray`.

  Show description for 76af644
  

  hawkinsp

  authored and

  copybara-github

  committed
  76af644

  Copy full SHA for 76af644

Commits on Aug 2, 2023

• Internal change.

  Show description for 95b7cfe
  

  Acme Contributor

  authored and

  copybara-github

  committed
  95b7cfe

  Copy full SHA for 95b7cfe

Commits on Jul 12, 2023

• Use default_factory instead of an instance for this dataclass field's default value.

  Show description for fb87fd5
  

  yilei

  authored and

  copybara-github

  committed
  fb87fd5

  Copy full SHA for fb87fd5

Commits on Jul 5, 2023

• Remove tfp.experimental.as_composite, as it does nothing (has been stubbed out).

  Show description for 36798a0
  

  SiegeLordEx

  authored and

  copybara-github

  committed
  36798a0

  Copy full SHA for 36798a0

Commits on Jun 13, 2023

• [NumPy] Fix uses of functions deprecated in NumPy 1.25.

  Show description for 546a47a
  

  hawkinsp

  authored and

  copybara-github

  committed
  546a47a

  Copy full SHA for 546a47a

Commits on Apr 26, 2023

• Silence some pytype errors.

  Show description for 98c4204
  

  rchen152

  authored and

  copybara-github

  committed
  98c4204

  Copy full SHA for 98c4204

Commits on Mar 28, 2023

• Fix pytype failures related to teaching pytype about NumPy scalar types.

  Show description for d1e69c9
  

  hawkinsp

  authored and

  copybara-github

  committed
  d1e69c9

  Copy full SHA for d1e69c9

Commits on Mar 23, 2023

• Suppress some pytype errors related to improved JAX types under pytype.

  Show description for ce6e9e4
  

  hawkinsp

  authored and

  copybara-github

  committed
  ce6e9e4

  Copy full SHA for ce6e9e4

Commits on Mar 17, 2023

• [JAX] Replace uses of jax.xla.Device with jax.Device.

  Show description for 0f5da20
  

  hawkinsp

  authored and

  copybara-github

  committed
  0f5da20

  Copy full SHA for 0f5da20

• Python 3.7 is needed for the OSSed Acme Colab

  Show description for 75e5523
  

  qstanczyk

  authored and

  copybara-github

  committed
  75e5523

  Copy full SHA for 75e5523

Commits on Mar 16, 2023

• Use tfp.experimental.auto_composite_tensor rather than tfp.experimental.register_composite.

  Show description for 337a456
  

  SiegeLordEx

  authored and

  copybara-github

  committed
  337a456

  Copy full SHA for 337a456

Commits on Mar 11, 2023

• Replaces references to jax.numpy.DeviceArray with jax.Array.

  Show description for 99c4977
  

  hawkinsp

  authored and

  copybara-github

  committed
  99c4977

  Copy full SHA for 99c4977

Commits on Mar 9, 2023

• Suppress some pytype errors related to jnp.DeviceArray == jax.Array.

  Show description for b9cb849
  

  hawkinsp

  authored and

  copybara-github

  committed
  b9cb849

  Copy full SHA for b9cb849

Commits on Mar 8, 2023

• Fix some pytype errors related to jnp.DeviceArray == jax.Array.

  Show description for bfc7a40
  

  hawkinsp

  authored and

  copybara-github

  committed
  bfc7a40

  Copy full SHA for bfc7a40

Commits on Mar 6, 2023

• Internal change

  Show description for 3626a94
  

  superbobry

  authored and

  copybara-github

  committed
  3626a94

  Copy full SHA for 3626a94

Pagination

Previous

Next