Commits

Branch selector

 master

User selector

All users

All time

Commit History

Commits on Feb 16, 2026

• Remove video wrapper to avoid directly calling ffmpeg.

  Show description for cafbfc5
  

  mwhoffman

  authored and

  copybara-github

  committed
  cafbfc5

  Copy full SHA for cafbfc5

• Remove video wrapper to avoid directly calling ffmpeg.

  Show description for 5a1da00
  

  Acme Contributor

  authored and

  copybara-github

  committed
  5a1da00

  Copy full SHA for 5a1da00

• Remove video wrapper to avoid directly calling ffmpeg.

  Show description for 15bd3e8
  

  mwhoffman

  authored and

  copybara-github

  committed
  15bd3e8

  Copy full SHA for 15bd3e8

Commits on Feb 2, 2026

• [pmap] Handle edge cases in get_from_first_device for jax_pmap_shmap_merge.

  Show description for e210507
  

  danielsuo

  authored and

  copybara-github

  committed
  e210507

  Copy full SHA for e210507

Commits on Jan 14, 2026

• Replace unicode escaped characters in ipynb files

  Show description for 05ced10
  

  AleksMat

  authored and

  copybara-github

  committed
  05ced10

  Copy full SHA for 05ced10

Commits on Dec 25, 2025

• [pmap] Avoid degraded performance under the new `jax.pmap`.

  Show description for c292597
  

  danielsuo

  authored and

  copybara-github

  committed
  c292597

  Copy full SHA for c292597

Commits on Dec 2, 2025

• [acme] Remove `jax_pmap_shmap_merge=False`.

  Show description for b7a0c92
  

  danielsuo

  authored and

  copybara-github

  committed
  b7a0c92

  Copy full SHA for b7a0c92

Commits on Sep 26, 2025

• [acme] Explicitly set `jax_pmap_shmap_merge=False`.

  Show description for 01b3696
  

  danielsuo

  authored and

  copybara-github

  committed
  01b3696

  Copy full SHA for 01b3696

Commits on Jul 22, 2025

• Add an optional customizable loss callable in the PPONetworks, whose output will be added to total loss.

  Show description for 8ec1a77
  

  Allen Wang

  authored and

  copybara-github

  committed
  8ec1a77

  Copy full SHA for 8ec1a77

Commits on Jul 16, 2025

• Silence pytype errors related to improved `jax.nn` type annotations.

  Show description for 6828035
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  6828035

  Copy full SHA for 6828035

Commits on May 29, 2025

• Actually include all WPO files this time

  Show description for 0be0d4f
  

  Acme Contributor

  authored and

  copybara-github

  committed
  0be0d4f

  Copy full SHA for 0be0d4f

Commits on May 3, 2025

• Add Wasserstein Policy Optimization (WPO, http://arxiv.org/pdf/2505.00663) to Acme

  Show description for ed4e57a
  

  Acme Contributor

  authored and

  copybara-github

  committed
  ed4e57a

  Copy full SHA for ed4e57a

Commits on Apr 30, 2025

• Automated Code Change

  Show description for 7578c75
  

  h-joo

  authored and

  copybara-github

  committed
  7578c75

  Copy full SHA for 7578c75

Commits on Apr 18, 2025

• Configure tests to explicitly use jax_threefry_partitionable=False.

  Show description for d920840
  

  jburnim

  authored and

  copybara-github

  committed
  d920840

  Copy full SHA for d920840

Commits on Jan 14, 2025

• jax.numpy.clip: update use of deprecated arguments.

  Show description for eedf63c
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  eedf63c

  Copy full SHA for eedf63c

Commits on Oct 30, 2024

• Added pytype None check to agent.py

  Show description for 5e7503f
  

  Acme Contributor

  authored and

  copybara-github

  committed
  5e7503f

  Copy full SHA for 5e7503f

Commits on Sep 17, 2024

• This CL updates the `Snapshotter` class to allow `None` as a valid value for `snapshot_ttl_seconds`. This TTL value is passed to `paths.process_path` which already supports a `None` TTL.

  Show description for 766a98b
  

  Acme Contributor

  authored and

  copybara-github

  committed
  766a98b

  Copy full SHA for 766a98b

Commits on Aug 26, 2024

• [pmap no rank reduce cleanup]: Prepare for flipping the

  Show description for 8c2a8c8
  

  pschuh

  authored and

  copybara-github

  committed
  8c2a8c8

  Copy full SHA for 8c2a8c8

Commits on Aug 13, 2024

• [numpy] Fix test failures under NumPy 2.0.

  Show description for d9c4319
  

  hawkinsp

  authored and

  copybara-github

  committed
  d9c4319

  Copy full SHA for d9c4319

Commits on Jul 16, 2024

• Fix the check for jax.Array.

  Show description for 8d76656
  

  sinopalnikov

  authored and

  copybara-github

  committed
  8d76656

  Copy full SHA for 8d76656

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

Pagination

Previous

Next