[WIP] Refactor codebase from JAX to PyTorch - #2

[WIP] Refactor codebase from JAX to PyTorch#2

nicosuter with Copilot wants to merge 1 commit into

mainETHRoboticsClub/openpi:mainfrom

copilot/refactor-jax-to-pytorchETHRoboticsClub/openpi:copilot/refactor-jax-to-pytorchCopy head branch name to clipboard