Prevent PI0/PI05 JAX to PyTorch weight precision drift.#978

Greyman-Seu wants to merge 2 commits into

Physical-Intelligence:mainPhysical-Intelligence/openpi:mainfrom

Greyman-Seu:feature-align-torch-jax-paramGreyman-Seu/openpi:feature-align-torch-jax-paramCopy head branch name to clipboard