Skip to content

Pull requests: google/brax

New pull requestNew
20 Open 103 Closed
20 Open 103 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviewsReview requiredApproved reviewChanges requested
Assignee
Filter by who’s assigned
Assigned to nobodyLoading
Sort
Sort by
NewestOldestMost commentedLeast commentedRecently updatedLeast recently updatedBest match
Most reactions
👍👎😄🎉😕❤️🚀👀

Pull requests list

Fix PPO input sharding to match pmap axis
#674 opened Jul 21, 2026 by lmj0209Loading…
Fix EpisodeWrapper dropping sub-step metrics during action_repeat (#610)
#670 opened Apr 17, 2026 by SAY-5Loading…
1
Fix SAC checkpoint restore device axis mismatch (#659)
#669 opened Apr 17, 2026 by SAY-5Loading…
1
Warn when init_noise_std and related params are ignored with tanh_normal (#661)
#668 opened Apr 16, 2026 by SAY-5Loading…
1
Security: Remediation of RCE via Insecure Pickle Deserialization in Model I/O
#667 opened Apr 5, 2026 by JoshuaProvosteLoading…
4 tasks done
Upgrade GitHub Actions for Node 24 compatibility
#666 opened Mar 11, 2026 by salmanmkcLoading…
Fix checkpoint.load_config
#665 opened Mar 9, 2026 by fengzileeeLoading…
1
ppo: add entropy cost annealing (linear + cosine schedules)
#660 opened Feb 26, 2026 by PhysicistJohnLoading…
3
math: scan solve_pgs iterations
#658 opened Jan 24, 2026 by chai1anyLoading…
simple history encoder for brax
#651 opened Dec 2, 2025 by samindaaLoading…
1
Fix #647 gradient_update_fn compatible with AdamW
#648 opened Nov 25, 2025 by ShanningZhuangLoading…
1
Add _get_obs to fast environment
#644 opened Oct 15, 2025 by YannBerthelotLoading…
Store terminal observations in AutoResetWrapper infos for value bootstrap
#641 opened Sep 25, 2025 by YannBerthelotLoading…
2
policy_params_fn
#636 opened Sep 9, 2025 by jameshealdLoading…
Add real-time rendering callback
#634 opened Aug 30, 2025 by bsyhLoading…
15
Add support for initial alpha value for SAC
#598 opened Apr 18, 2025 by varunagrawalLoading…
6
TD3 baseline
#499 opened Jul 5, 2024 by anasbekheitLoading…
@btaba
1
Switch gym dependency to gymnasium
#486 opened May 14, 2024 by giotherobotLoading…
9
io/image: Replace PyTinyrenderer with a Pure JAX renderer
#367 opened Jun 2, 2023 by JoeyTengLoading…
7
Agents: Short-Horizon Actor Critic
#262 opened Nov 19, 2022 by peabody124ContributorLoading…
16
ProTip! Follow long discussions with comments:>50.