chore: update neuron dependencies

Author: web-flow

test/images/neuron-inference/Dockerfile

@@ -11,11 +11,11 @@ ENV PYTHONUNBUFFERED=1
 
 # Neuron SDK components version numbers
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/inference/2.5.1/Dockerfile.neuronx
-ARG NEURONX_CC_VERSION=2.22.12471.0+b4a00d10
-ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.11.19912+e48cd891
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.29.41.0-681fef5f5
-ARG NEURONX_RUNTIME_LIB_VERSION=2.29.40.0-f954cd7a5
-ARG NEURONX_TOOLS_VERSION=2.27.33.0-5d9c0b901
+ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22
+ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186
+ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce
+ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42
 
 # Python
 ARG PYTHON=python3.10

test/images/neuron-training/Dockerfile

@@ -7,11 +7,11 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 # Neuron SDK component versions (pin these precisely)
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/training/2.5.1/Dockerfile.neuronx
-ARG NEURONX_CC_VERSION=2.22.12471.0+b4a00d10
-ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.11.19912+e48cd891
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.29.41.0-681fef5f5
-ARG NEURONX_RUNTIME_LIB_VERSION=2.29.40.0-f954cd7a5
-ARG NEURONX_TOOLS_VERSION=2.27.33.0-5d9c0b901
+ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22
+ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186
+ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce
+ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42
 
 # Python
 ARG PYTHON=python3.10

test/images/neuron/Dockerfile

@@ -2,12 +2,12 @@ FROM public.ecr.aws/docker/library/ubuntu:22.04
 
 # Neuron SDK components version numbers
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/training/2.5.1/Dockerfile.neuronx
-ARG NEURONX_DISTRIBUTED_VERSION=0.16.25997+f431c02e
-ARG NEURONX_CC_VERSION=2.22.12471.0+b4a00d10
-ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.11.19912+e48cd891
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.29.41.0-681fef5f5
-ARG NEURONX_RUNTIME_LIB_VERSION=2.29.40.0-f954cd7a5
-ARG NEURONX_TOOLS_VERSION=2.27.33.0-5d9c0b901
+ARG NEURONX_DISTRIBUTED_VERSION=0.18.27753+1cafd54f
+ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22
+ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186
+ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce
+ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42
 
 ARG PYTHON=python3.10
 ARG PYTHON_VERSION=3.10.12