Skip to content

Fix PPO input sharding to match pmap axis - #674

Open
lmj0209 wants to merge 1 commit into
google:mainfrom
lmj0209:fix/ppo-jax-device-put
Open

Fix PPO input sharding to match pmap axis#674
lmj0209 wants to merge 1 commit into
google:mainfrom
lmj0209:fix/ppo-jax-device-put

Fix PPO input sharding to match pmap axis

0802964
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 21, 2026 in 8s

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

✅ 0802964 Author: @lmj0209 <97*****06​@qq.com>

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

View more details on Google CLA