We are working on the Helion TPU/Pallas backend.
- Local pytorch checkout: ~/pytorch
- Local helion checkout: ~/helion
- TPU access:
kubectl exec -it <pod> -- /bin/bash(default pod:<username>-torchtpu) - On the TPU pod, activate the venv first:
source ~/.venv/bin/activate