Fix torch dependency versioning

See keras-team/keras#19603 for reason why we needed to pin this

README.md

@@ -100,10 +100,10 @@ Attempting to install the `torch` backend using
 poetry install --with torch
 ```
 
-fails with a message like `Unable to find installation candidates for torch (2.3.1+cpu)`. To remedy, run
+fails with a message like `Unable to find installation candidates for torch (2.2.2+cpu)`. To remedy, run
 
 ```bash
-pip install torch torchaudio torchvision
+pip install torch~=2.2.2 torchvision~=0.17.2 torchaudio~=2.2.2
 ```
 
 <!-- end installation -->

pyproject.toml

@@ -82,17 +82,17 @@ tensorflow = { version = "~2.16.1", extras = ["and-cuda"] }
 optional = true
 
 [tool.poetry.group.torch.dependencies]
-torch = { version = "~2.3.1+cpu", source = "pytorch-cpu" }
-torchvision = { version = "~0.18.1+cpu", source = "pytorch-cpu" }
-torchaudio = { version = "~2.3.1+cpu", source = "pytorch-cpu" }
+torch = { version = "~2.2.2+cpu", source = "pytorch-cpu" }
+torchvision = { version = "~0.17.2+cpu", source = "pytorch-cpu" }
+torchaudio = { version = "~2.2.2+cpu", source = "pytorch-cpu" }
 
 [tool.poetry.group.torch-gpu]
 optional = true
 
 [tool.poetry.group.torch-gpu.dependencies]
-torch = "~2.3.1+cpu"
-torchvision = "~0.18.1+cpu"
-torchaudio = "~2.3.1+cpu"
+torch = "~2.2.2"
+torchvision = "~0.17.2"
+torchaudio = "~2.2.2"
 
 [tool.poetry.group.jax]
 optional = true