Commits

Breadcrumbs

History for

onmain

User selector

All users

All time

Commit History

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 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

• 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

• 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 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 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

Commits on Dec 7, 2025

• Fix counting issue in running statistics.

  Show description for ca7649c
  

  btaba

  authored and

  copybara-github

  committed
  ca7649c

  Copy full SHA for ca7649c

Commits on Nov 25, 2025

• Hook into per_step metrics for training.

  Show description for 12c29bd
  

  btaba

  authored and

  copybara-github

  committed
  12c29bd

  Copy full SHA for 12c29bd

Commits on Nov 21, 2025

• Add sps to training metric logger.

  Show description for 853f207
  

  btaba

  authored and

  copybara-github

  committed
  853f207

  Copy full SHA for 853f207

Commits on Nov 10, 2025

• Add Mish activation function.

  Show description for d96ab73
  

  Brax Team

  authored and

  copybara-github

  committed
  d96ab73

  Copy full SHA for d96ab73

Commits on Nov 7, 2025

• Remove ppo test assertion, only increments during tracing and is error prone.

  Show description for 9b7af2a
  

  btaba

  authored and

  copybara-github

  committed
  9b7af2a

  Copy full SHA for 9b7af2a

• Ignore pytype errors produced with --use-functools-partial-overlay

  Show description for 1a6b943
  

  superbobry

  authored and

  copybara-github

  committed
  1a6b943

  Copy full SHA for 1a6b943

• Fix enum member py version issue.

  Show description for 124bf39
  

  btaba

  authored and

  copybara-github

  committed
  124bf39

  Copy full SHA for 124bf39

Commits on Oct 30, 2025

• Fix reset function call in PPO training

  Michael-Beukman

  authored
  8e85a53

  Copy full SHA for 8e85a53

Commits on Oct 20, 2025

• Prepare for Python 3.13: wrap `functools.partial` enum members into `enum.member()`

  Show description for eb9a54a
  

  oprypin

  authored and

  copybara-github

  committed
  eb9a54a

  Copy full SHA for eb9a54a

Commits on Oct 7, 2025

• Add donate_argnums to brax PPO. Avoids graph recaptures with MJX-Warp.

  Show description for dcab2b4
  

  btaba

  authored and

  copybara-github

  committed
  dcab2b4

  Copy full SHA for dcab2b4