Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Mar 30, 2026

• [JAX] Replace jnp.clip(..., a_min=..., a_max=...) with jnp.clip(..., min=..., max=...).

  Show description for 5c994de
  

  hawkinsp

  authored and

  copybara-github

  committed
  5c994de

  Copy full SHA for 5c994de

Commits on Mar 25, 2026

• [pmap] In-line definitions of `jax.device_put_sharded` and `jax.device_put_replicated`.

  Show description for 303d89d
  

  danielsuo

  authored and

  copybara-github

  committed
  303d89d

  Copy full SHA for 303d89d

Commits on Mar 16, 2026

• Prepare mujoco playground for release.

  Show description for ae5bbb0
  

  btaba

  authored and

  copybara-github

  committed
  ae5bbb0

  Copy full SHA for ae5bbb0

Commits on Mar 15, 2026

• Update Brax version to 0.14.2.

  Show description for f489566
  

  Brax Team

  authored and

  copybara-github

  committed
  f489566

  Copy full SHA for f489566

Commits on Mar 14, 2026

• Use GAE returns as the target for the value function loss in PPO.

  Show description for 3d4b070
  

  Brax Team

  authored and

  copybara-github

  committed
  3d4b070

  Copy full SHA for 3d4b070

Commits on Mar 11, 2026

• Add distributional critic to PPO and enhance observation normalization.

  Show description for 2b109a0
  

  Brax Team

  authored and

  copybara-github

  committed
  2b109a0

  Copy full SHA for 2b109a0

Commits on Mar 9, 2026

• Update hypers for cartpole vision.

  Show description for 91fc712
  

  btaba

  authored and

  copybara-github

  committed
  91fc712

  Copy full SHA for 91fc712

• Fix Brax ES argsort failure by resharding weights before fitness shaping

  Show description for 5433c22
  

  btaba

  authored and

  copybara-github

  committed
  5433c22

  Copy full SHA for 5433c22

Commits on Mar 6, 2026

• Add CNN kernel init and spatial softmax (h/t zakka)

  Show description for 4597b9c
  

  btaba

  authored and

  copybara-github

  committed
  4597b9c

  Copy full SHA for 4597b9c

Commits on Mar 4, 2026

• Update Python CI versions (#663)

  Show description for d2cb645
  btaba

  committed
  d2cb645

  Copy full SHA for d2cb645

• Import Brax PR #662: Vision update

  Show description for 196d641
  

  btaba

  authored and

  copybara-github

  committed
  196d641

  Copy full SHA for 196d641

• Import Brax PR #662: Vision update

  Show description for e183ac6
  

  btaba

  authored and

  copybara-github

  committed
  e183ac6

  Copy full SHA for e183ac6

Commits on Mar 3, 2026

• Fix Brax external tests by updating jax.sharding.PartitionSpec

  Show description for 71c94d9
  

  btaba

  authored and

  copybara-github

  committed
  71c94d9

  Copy full SHA for 71c94d9

• Vision update (#662)

  Show description for 1a67c0c
  

  StafaH

  authored and

  btaba

  committed
  1a67c0c

  Copy full SHA for 1a67c0c

Commits on Feb 26, 2026

• [pmap] Remove `jax.config.pmap_shmap_merge`.

  Show description for 7120ee3
  

  danielsuo

  authored and

  copybara-github

  committed
  7120ee3

  Copy full SHA for 7120ee3

Commits on Feb 12, 2026

• Bump brax to 0.14.1

  Show description for 6da6b4f
  

  Brax Team

  authored and

  copybara-github

  committed
  6da6b4f

  Copy full SHA for 6da6b4f

• Fix for brax es train, which was broken by cl/869007637.

  Show description for f1f29c3
  

  Brax Team

  authored and

  copybara-github

  committed
  f1f29c3

  Copy full SHA for f1f29c3

Commits on Feb 10, 2026

• Fix checkpoint.save

  Show description for 770e837
  

  thowell

  authored and

  copybara-github

  committed
  770e837

  Copy full SHA for 770e837

Commits on Feb 9, 2026

• [pmap] Prepare brax ES agent for jax_pmap_shmap_merge=True.

  Show description for aed6acc
  

  danielsuo

  authored and

  copybara-github

  committed
  aed6acc

  Copy full SHA for aed6acc

Commits on Feb 6, 2026

• Check more general exception type in assert.

  Show description for 8b83534
  

  erikfrey

  authored and

  copybara-github

  committed
  8b83534

  Copy full SHA for 8b83534

Commits on Jan 29, 2026

• Add mean_kernel_init_fn to checkpoint serialization keywords

  Show description for 6456720
  

  btaba

  authored and

  copybara-github

  committed
  6456720

  Copy full SHA for 6456720

• Flatten params.

  Show description for 8e4b482
  

  btaba

  authored and

  copybara-github

  committed
  8e4b482

  Copy full SHA for 8e4b482

• Add soft-sign clipping (mean_clip_scale) and configurable mean_kernel_init to PolicyModuleWithStd

  Show description for 8578b9c
  

  btaba

  authored and

  copybara-github

  committed
  8578b9c

  Copy full SHA for 8578b9c

Commits on Jan 14, 2026

• Replace unicode escaped characters in ipynb files

  Show description for 5294163
  

  AleksMat

  authored and

  copybara-github

  committed
  5294163

  Copy full SHA for 5294163

Commits on Jan 2, 2026

• Import https://github.com/google/brax/pull/654

  Show description for 5abb003
  

  btaba

  authored and

  copybara-github

  committed
  5abb003

  Copy full SHA for 5abb003

• Merge branch 'main' of https://github.com/google/brax

  Show description for 9f59e80
  btaba

  committed
  9f59e80

  Copy full SHA for 9f59e80

• Fix episode metrics aggregation in EpisodeWrapper (#654)

  ak24watch

  authored
  ef599ad

  Copy full SHA for ef599ad

Commits on Dec 29, 2025

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

  Show description for 5b3c516
  

  danielsuo

  authored and

  copybara-github

  committed
  5b3c516

  Copy full SHA for 5b3c516

Commits on Dec 16, 2025

• Update pypi.yml

  btaba

  authored
  f22fd37

  Copy full SHA for f22fd37

• Bump brax to 0.14.0

  Show description for 69f0023
  

  btaba

  authored and

  copybara-github

  committed
  69f0023

  Copy full SHA for 69f0023

Commits on Dec 15, 2025

• Add clipped value loss.

  Show description for f9ba54f
  

  btaba

  authored and

  copybara-github

  committed
  f9ba54f

  Copy full SHA for f9ba54f

• Bootstrap on timeout.

  Show description for e8c9c18
  

  btaba

  authored and

  copybara-github

  committed
  e8c9c18

  Copy full SHA for e8c9c18

Commits on Dec 11, 2025

• Apply adaptive KL LR before grad.

  Show description for a6b0c6b
  

  btaba

  authored and

  copybara-github

  committed
  a6b0c6b

  Copy full SHA for a6b0c6b

Commits on Dec 9, 2025

• Fix steps_between_logging.

  Show description for 5d3c5e4
  

  btaba

  authored and

  copybara-github

  committed
  5d3c5e4

  Copy full SHA for 5d3c5e4

Commits on Dec 8, 2025

• EMA running stats.

  Show description for 74c90a2
  

  btaba

  authored and

  copybara-github

  committed
  74c90a2

  Copy full SHA for 74c90a2