chore!: change folder structure

.github/workflows/build-neuron-jax-training-dlc.yaml

@@ -4,14 +4,14 @@ on:
   push:
     branches: [ main ]
     paths:
-      - 'docker/jax/training/**'
-      - 'docker/common/**'
+      - 'jax/training/**'
+      - 'common/**'
       - '.github/workflows/build-neuron-jax-training-dlc.yaml'
   pull_request:
     branches: [ main ]        
     paths:
-      - 'docker/jax/training/**'
-      - 'docker/common/**'
+      - 'jax/training/**'
+      - 'common/**'
   workflow_dispatch:    
     inputs:
         jax_versions:
@@ -46,7 +46,7 @@ jobs:
               version=$(echo "$version" | xargs) # trim whitespace
               # Validate version format (only allow alphanumeric, dots, and hyphens)
               if [[ "$version" =~ ^[a-zA-Z0-9.-]+$ ]]; then
-                path="docker/jax/training/$version"
+                path="jax/training/$version"
                 if [ -f "$path/Dockerfile.neuronx" ]; then
                   matrix_json=$(echo "$matrix_json" | jq --arg v "$version" --arg p "$path" '. + [{"version": $v, "path": $p}]')
                 fi
@@ -55,7 +55,7 @@ jobs:
           else
             # Detect changed Dockerfile paths
             changed_paths=$(git diff --name-only "origin/$BASE_REF" "$SHA" \
-              | grep '^docker/jax/training/.*/Dockerfile.neuronx$' \
+              | grep '^jax/training/.*/Dockerfile.neuronx$' \
               | xargs -n1 dirname \
               | sort -u)
 
@@ -96,7 +96,7 @@ jobs:
         env:
           MATRIX_PATH: ${{ matrix.path }}
         run: |
-          cp -r docker/common/* "$MATRIX_PATH"/
+          cp -r common/* "$MATRIX_PATH"/
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3

.github/workflows/build-neuron-pytorch-inference-dlc.yaml

@@ -4,14 +4,14 @@ on:
   push:
     branches: [ main ]
     paths:
-      - 'docker/pytorch/inference/**'
-      - 'docker/common/**'
+      - 'pytorch/inference/**'
+      - 'common/**'
       - '.github/workflows/build-neuron-pytorch-inference-dlc.yaml'
   pull_request:
     branches: [ main ]        
     paths:
-      - 'docker/pytorch/inference/**'
-      - 'docker/common/**'
+      - 'pytorch/inference/**'
+      - 'common/**'
   workflow_dispatch:    
     inputs:
         pytorch_versions:
@@ -46,7 +46,7 @@ jobs:
               version=$(echo "$version" | xargs) # trim whitespace
               # Validate version format (only allow alphanumeric, dots, and hyphens)
               if [[ "$version" =~ ^[a-zA-Z0-9.-]+$ ]]; then
-                path="docker/pytorch/inference/$version"
+                path="pytorch/inference/$version"
                 if [ -f "$path/Dockerfile.neuronx" ]; then
                   matrix_json=$(echo "$matrix_json" | jq --arg v "$version" --arg p "$path" '. + [{"version": $v, "path": $p}]')
                 fi
@@ -55,7 +55,7 @@ jobs:
           else
             # Detect changed Dockerfile paths
             changed_paths=$(git diff --name-only "origin/$BASE_REF" "$SHA" \
-              | grep '^docker/pytorch/inference/.*/Dockerfile.neuronx$' \
+              | grep '^pytorch/inference/.*/Dockerfile.neuronx$' \
               | xargs -n1 dirname \
               | sort -u)
 
@@ -96,7 +96,7 @@ jobs:
         env:
           MATRIX_PATH: ${{ matrix.path }}
         run: |
-          cp -r docker/common/* "$MATRIX_PATH"/
+          cp -r common/* "$MATRIX_PATH"/
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3

.github/workflows/build-neuron-pytorch-training-dlc.yaml

@@ -4,14 +4,14 @@ on:
   push:
     branches: [ main ]
     paths:
-      - 'docker/pytorch/training/**'
-      - 'docker/common/**'
+      - 'pytorch/training/**'
+      - 'common/**'
       - '.github/workflows/build-neuron-pytorch-training-dlc.yaml'
   pull_request:
     branches: [ main ]        
     paths:
-      - 'docker/pytorch/training/**'
-      - 'docker/common/**'
+      - 'pytorch/training/**'
+      - 'common/**'
   workflow_dispatch:    
     inputs:
         pytorch_versions:
@@ -46,7 +46,7 @@ jobs:
               version=$(echo "$version" | xargs) # trim whitespace
               # Validate version format (only allow alphanumeric, dots, and hyphens)
               if [[ "$version" =~ ^[a-zA-Z0-9.-]+$ ]]; then
-                path="docker/pytorch/training/$version"
+                path="pytorch/training/$version"
                 if [ -f "$path/Dockerfile.neuronx" ]; then
                   matrix_json=$(echo "$matrix_json" | jq --arg v "$version" --arg p "$path" '. + [{"version": $v, "path": $p}]')
                 fi
@@ -55,7 +55,7 @@ jobs:
           else
             # Detect changed Dockerfile paths
             changed_paths=$(git diff --name-only "origin/$BASE_REF" "$SHA" \
-              | grep '^docker/pytorch/training/.*/Dockerfile.neuronx$' \
+              | grep '^pytorch/training/.*/Dockerfile.neuronx$' \
               | xargs -n1 dirname \
               | sort -u)
 
@@ -96,7 +96,7 @@ jobs:
         env:
           MATRIX_PATH: ${{ matrix.path }}
         run: |
-          cp -r docker/common/* "$MATRIX_PATH"/
+          cp -r common/* "$MATRIX_PATH"/
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3

.github/workflows/build-neuron-vllm-inference-dlc.yaml

@@ -4,14 +4,14 @@ on:
   push:
     branches: [ main ]
     paths:
-      - 'docker/vllm/inference/**'
-      - 'docker/common/**'
+      - 'vllm/inference/**'
+      - 'common/**'
       - '.github/workflows/build-neuron-vllm-inference-dlc.yaml'
   pull_request:
     branches: [ main ]        
     paths:
-      - 'docker/vllm/inference/**'
-      - 'docker/common/**'
+      - 'vllm/inference/**'
+      - 'common/**'
   workflow_dispatch:    
     inputs:
         vllm_versions:
@@ -46,7 +46,7 @@ jobs:
               version=$(echo "$version" | xargs) # trim whitespace
               # Validate version format (only allow alphanumeric, dots, and hyphens)
               if [[ "$version" =~ ^[a-zA-Z0-9.-]+$ ]]; then
-                path="docker/vllm/inference/$version"
+                path="vllm/inference/$version"
                 if [ -f "$path/Dockerfile.neuronx" ]; then
                   matrix_json=$(echo "$matrix_json" | jq --arg v "$version" --arg p "$path" '. + [{"version": $v, "path": $p}]')
                 fi
@@ -55,7 +55,7 @@ jobs:
           else
             # Detect changed Dockerfile paths
             changed_paths=$(git diff --name-only "origin/$BASE_REF" "$SHA" \
-              | grep '^docker/vllm/inference/.*/Dockerfile.neuronx$' \
+              | grep '^vllm/inference/.*/Dockerfile.neuronx$' \
               | xargs -n1 dirname \
               | sort -u)
 
@@ -96,7 +96,7 @@ jobs:
         env:
           MATRIX_PATH: ${{ matrix.path }}
         run: |
-          cp -r docker/common/* "$MATRIX_PATH"/
+          cp -r common/* "$MATRIX_PATH"/
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3

README.md

@@ -14,7 +14,7 @@ To build a specific container, navigate to the repository root and use Docker bu
 
 ```bash
 # Example: Build PyTorch 2.7.0 inference container
-docker build -f docker/pytorch/inference/2.7.0/Dockerfile.neuronx -t my-neuron-container .
+docker build -f pytorch/inference/2.7.0/Dockerfile.neuronx -t my-neuron-container .
 ```
 
 ### Customizing Containers
@@ -79,4 +79,4 @@ See [SECURITY](SECURITY.md) for more information.
 
 ## License
 
-This project is licensed under the Apache-2.0 License.
+This project is licensed under the Apache-2.0 License.