Fix SAC checkpoint restore device axis mismatch (#659) - #669

Fix SAC checkpoint restore device axis mismatch (#659)#669

SAY-5 wants to merge 1 commit into

google:maingoogle/brax:mainfrom

SAY-5:fix-sac-checkpoint-device-axisSAY-5/brax:fix-sac-checkpoint-device-axisCopy head branch name to clipboard