Openpi

Writing some notes on setting this up for training:

uv run scripts/compute_norm_stats.py --config-name pi05_koch

XLA_PYTHON_CLIENT_MEM_FRACTION=0.9 uv run scripts/train.py pi05_koch --exp-name=my_experiment --overwrite

Troubleshooting

Had to downgrade