Skip to content

Preserve terminal observation in BraxAutoResetWrapper for truncation bootstrap - #320

Open
dparikh79 wants to merge 2 commits into
google-deepmind:mainfrom
dparikh79:fix/305-brax-autoreset-preserve-terminal-obs
Open

Preserve terminal observation in BraxAutoResetWrapper for truncation bootstrap#320
dparikh79 wants to merge 2 commits into
google-deepmind:mainfrom
dparikh79:fix/305-brax-autoreset-preserve-terminal-obs

Seed AutoResetWrapper_final_obs on reset for JAX tree consistency

ad0fd5e
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded May 16, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

ad0fd5e Author: @dparikh79 <dhru********h79​@gmail.com>

(Only the first commit for a unique contributor is listed.)