diff --git a/.github/workflows/code_style_fix.yaml b/.github/workflows/code_style_fix.yaml
new file mode 100644
index 0000000..a6d4788
--- /dev/null
+++ b/.github/workflows/code_style_fix.yaml
@@ -0,0 +1,65 @@
+name: Automatic Code Style Fix
+
+permissions: write-all
+
+
+on:
+  pull_request:
+    branches: [main]
+  workflow_call:
+    secrets:
+      github-token:
+        description: "if provided an user GH's token, it will push update; requires `push` event"
+        required: false
+
+jobs:
+  auto_code_style_fix:
+    runs-on: ubuntu-22.04
+    steps:
+      # - name: Checkout ️🛒
+      #   uses: actions/checkout@v4
+      #   with:
+      #     ref: ${{ github.head_ref }}
+      #     fetch-depth: 0
+
+      - name: Checkout 🛎️
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+          submodules: recursive
+          token: ${{ secrets.github-token || github.token }}
+
+      - name: Black ◼️
+        uses: psf/black@stable
+        with:
+          # options: "-l120 -C --target-version py38 chat eval finetune generate lit_gpt notebooks pretrain quantize scripts tests xla"
+          options: "-l120 -C --target-version py38 notebooks src"
+          jupyter: true
+
+      - name: Ruff ®️
+        uses: chartboost/ruff-action@v1
+        with:
+          args: --select E,W,F,S --extend-select C4,SIM,RET,PT,I001,ANN001,ANN201,ANN205,ANN206,ARG001 --ignore E501,E731,S108,S101,S113,S603,PT007,S310,E402,PT004,C408 --per-file-ignores "*/**/__init__.py":I001,"tests/*":ANN --line-length 120 --target-version py38 --fix-only
+
+      # - name: Commit files ✔️
+      #   run: |
+      #     git config --local user.email "github-actions[bot]@users.noreply.github.com"
+      #     git config --local user.name "github-actions[bot]"
+      #     git diff-index --quiet HEAD || git commit -a -m "Automatic code-style fix."
+
+      # - name: Pushing changes 🛠 ️
+      #   uses: ad-m/github-push-action@master
+      #   with:
+      #     branch: ${{ github.head_ref }}
+      #     github_token: ${{ secrets.github-token || github.token }}
+      - name: Fixing Pull Request ↩️
+        # if: always() && inputs.push-fixes == true && steps.precommit.outcome == 'failure'
+        uses: actions-js/push@v1.4
+        with:
+          # github_token: ${{ secrets.github-token || github.token }}
+          # github_token: ${{ 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 }}
+          # github_token: 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
+          # github_token: ${{ secrets.ruff_black_pat }}
+          github_token: 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
+          message: "pre-commit: running and fixing..."
+          branch: ${{ github.head_ref || github.ref_name }}
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index dbc5cdb..ce07736 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -5,6 +5,10 @@ on:
   push:
     branches: [main]
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}-${{ github.head_ref }}
+  cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
+
 jobs:
   test:
     strategy: