-f https://download.pytorch.org/whl/torch
torch==2.7.1 ; sys_platform == 'darwin'  # macOS
torch==2.7.1+cpu; sys_platform != 'darwin' # Non-macOS (CPU-only), like on TPU
yapf==0.40.2  # N.B.: keep in sync with `infra/ansible/config/pip.yaml`, `.github/workflows/lintercheck.yml`
flax==0.10.6
