Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

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

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

• Merge pull request #645 from Michael-Beukman:patch-1

  Show description for 84c7037
  copybara-github

  committed
  84c7037

  Copy full SHA for 84c7037

• 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

Commits on Oct 6, 2025

• Update for new pmap.

  Show description for 3085074
  

  btaba

  authored and

  copybara-github

  committed
  3085074

  Copy full SHA for 3085074

Commits on Sep 26, 2025

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

  Show description for ab34392
  

  danielsuo

  authored and

  copybara-github

  committed
  ab34392

  Copy full SHA for ab34392

Commits on Sep 23, 2025

• Fixes bug passing non-bool for `jnp.where`, which is no longer supported.

  Show description for 9f7c582
  

  erikfrey

  authored and

  copybara-github

  committed
  9f7c582

  Copy full SHA for 9f7c582

Commits on Sep 17, 2025

• adaptive kl learning rate

  Show description for 03a5090
  

  btaba

  authored and

  copybara-github

  committed
  03a5090

  Copy full SHA for 03a5090

Commits on Sep 16, 2025

• remove redundant call

  Show description for 4e60b25
  

  Brax Team

  authored and

  copybara-github

  committed
  4e60b25

  Copy full SHA for 4e60b25

Commits on Sep 15, 2025

• allow for per-step aggregation of metrics in the eval wrapper

  Show description for ed29545
  

  btaba

  authored and

  copybara-github

  committed
  ed29545

  Copy full SHA for ed29545

Commits on Sep 11, 2025

• vf_coeff and network initializers

  Show description for 202bbfa
  

  btaba

  authored and

  copybara-github

  committed
  202bbfa

  Copy full SHA for 202bbfa

Commits on Sep 9, 2025

• Make activation config more robust.

  Show description for 7428fd3
  

  btaba

  authored and

  copybara-github

  committed
  7428fd3

  Copy full SHA for 7428fd3

• Support custom activation for checkpointing.

  Show description for 31b247b
  

  btaba

  authored and

  copybara-github

  committed
  31b247b

  Copy full SHA for 31b247b

Commits on Aug 21, 2025

• Pass `torch.Tensor` to `jax.dlpack.from_dlpack` directly

  Show description for 338402c
  

  superbobry

  authored and

  copybara-github

  committed
  338402c

  Copy full SHA for 338402c

Commits on Aug 19, 2025

• Updates to Playground for warp.

  Show description for 75cddee
  

  btaba

  authored and

  copybara-github

  committed
  75cddee

  Copy full SHA for 75cddee

Commits on Aug 15, 2025

• Bump brax to 0.13.0 in light of orbax-checkpoint.

  Show description for 6345a3b
  

  btaba

  authored and

  copybara-github

  committed
  6345a3b

  Copy full SHA for 6345a3b

Commits on Aug 14, 2025

• Bump orbax-checkpoint due to breaking change.

  Show description for 0c223b1
  

  btaba

  authored and

  copybara-github

  committed
  0c223b1

  Copy full SHA for 0c223b1

Pagination

Previous

Next