Update rust.yml

.github/workflows/engine.yml

@@ -30,7 +30,7 @@ jobs:
           persist-credentials: false
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
           cache: false
@@ -42,7 +42,7 @@ jobs:
         run: rustup target add wasm32-unknown-unknown
 
       - name: Add rust-src
-        run: rustup component add rust-src --toolchain nightly-2025-06-16-x86_64-unknown-linux-gnu
+        run: rustup component add rust-src --toolchain nightly-2025-05-01-x86_64-unknown-linux-gnu
       - name: Install binaryen
         run: |
           set -e

.github/workflows/large-tests.yml

@@ -15,7 +15,7 @@ jobs:
           persist-credentials: false
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - name: nanoGPT Mock

.github/workflows/pypi.yml

@@ -50,7 +50,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
           cache: false
@@ -116,7 +116,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
           cache: false

.github/workflows/release.yml

@@ -50,7 +50,7 @@ jobs:
     steps:
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48   #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
           cache: false
@@ -118,27 +118,27 @@ jobs:
         include:
           - build: windows-msvc
             os: windows-latest
-            rust: nightly-2025-06-16
+            rust: nightly-2025-05-01
             target: x86_64-pc-windows-msvc
           - build: macos
             os: macos-13
-            rust: nightly-2025-06-16
+            rust: nightly-2025-05-01
             target: x86_64-apple-darwin
           - build: macos-aarch64
             os: macos-13
-            rust: nightly-2025-06-16
+            rust: nightly-2025-05-01
             target: aarch64-apple-darwin
           - build: linux-musl
-            os: ubuntu-latest-64-cores
-            rust: nightly-2025-06-16
+            os: ubuntu-22.04
+            rust: nightly-2025-05-01
             target: x86_64-unknown-linux-musl
           - build: linux-gnu
-            os: ubuntu-latest-64-cores
-            rust: nightly-2025-06-16
+            os: ubuntu-22.04
+            rust: nightly-2025-05-01
             target: x86_64-unknown-linux-gnu
           - build: linux-aarch64
-            os: ubuntu-latest-64-cores
-            rust: nightly-2025-06-16
+            os: ubuntu-22.04
+            rust: nightly-2025-05-01
             target: aarch64-unknown-linux-gnu
 
     steps:
@@ -200,7 +200,7 @@ jobs:
 
       - name: Build release binary (asm)
         if: matrix.build == 'linux-gnu'
-        run: ${{ env.CARGO }} build --release ${{ env.TARGET_FLAGS }} -Z sparse-registry --features asm,mimalloc
+        run: ${{ env.CARGO }} build --release ${{ env.TARGET_FLAGS }} -Z sparse-registry --features mimalloc
 
       - name: Build release binary (metal)
         if: matrix.build == 'macos-aarch64'

.github/workflows/rust.yml

@@ -33,7 +33,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3
@@ -57,7 +57,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - name: Build
@@ -75,7 +75,7 @@ jobs:
           persist-credentials: false
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - name: Docs
@@ -93,7 +93,7 @@ jobs:
           persist-credentials: false
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -118,7 +118,7 @@ jobs:
   #   persist-credentials: false
   #     - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
   #       with:
-  #         toolchain: nightly-2025-06-16
+  #         toolchain: nightly-2025-05-01
   #         override: true
   #         components: rustfmt, clippy
   #     - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -144,7 +144,7 @@ jobs:
   ultra-overflow-tests_og-lookup:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
     steps:
@@ -154,7 +154,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -180,7 +180,7 @@ jobs:
   ultra-overflow-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
 
@@ -192,7 +192,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
 
@@ -231,7 +231,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -258,7 +258,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: jetli/wasm-pack-action@0d096b08b4e5a7de8c28de67e11e945404e9eefa  #v0.4.0
@@ -271,7 +271,7 @@ jobs:
       - name: Install wasm32-unknown-unknown
         run: rustup target add wasm32-unknown-unknown
       - name: Add rust-src
-        run: rustup component add rust-src --toolchain nightly-2025-06-16-x86_64-unknown-linux-gnu
+        run: rustup component add rust-src --toolchain nightly-2025-05-01-x86_64-unknown-linux-gnu
       - name: Create webdriver.json to disable timeouts
         run: |
           echo '{"args": ["--headless", "--disable-gpu", "--disable-dev-shm-usage", "--no-sandbox"]}' > webdriver.json
@@ -286,7 +286,7 @@ jobs:
   mock-proving-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
 
@@ -297,7 +297,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -356,7 +356,7 @@ jobs:
   prove-and-verify-evm-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     # needs: [build, library-tests, docs, python-tests, python-integration-tests]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
@@ -370,7 +370,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: jetli/wasm-pack-action@0d096b08b4e5a7de8c28de67e11e945404e9eefa  #v0.4.0
@@ -394,7 +394,7 @@ jobs:
           node-version: "18.12.1"
           cache: "pnpm"
       - name: "Add rust-src"
-        run: rustup component add rust-src --toolchain nightly-2025-06-16-x86_64-unknown-linux-gnu
+        run: rustup component add rust-src --toolchain nightly-2025-05-01-x86_64-unknown-linux-gnu
       - name: Install dependencies for js tests and package
         run: |
           pnpm install --frozen-lockfile
@@ -433,7 +433,7 @@ jobs:
   #         persist-credentials: false
   #     - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
   #       with:
-  #         toolchain: nightly-2025-06-16
+  #         toolchain: nightly-2025-05-01
   #         override: true
   #         components: rustfmt, clippy
   #     - uses: jetli/wasm-pack-action@0d096b08b4e5a7de8c28de67e11e945404e9eefa  #v0.4.0
@@ -441,7 +441,7 @@ jobs:
   #         # Pin to version 0.12.1
   #         version: 'v0.12.1'
   #     - name: Add rust-src
-  #       run: rustup component add rust-src --toolchain nightly-2025-06-16
+  #       run: rustup component add rust-src --toolchain nightly-2025-05-01
   #     - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  #v4.2.2
   #       with:
   #         persist-credentials: false
@@ -459,7 +459,7 @@ jobs:
   prove-and-verify-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     needs: [build, library-tests, docs]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
@@ -473,7 +473,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: jetli/wasm-pack-action@0d096b08b4e5a7de8c28de67e11e945404e9eefa  #v0.4.0
@@ -484,7 +484,7 @@ jobs:
         run: rustup target add wasm32-unknown-unknown
 
       - name: Add rust-src
-        run: rustup component add rust-src --toolchain nightly-2025-06-16-x86_64-unknown-linux-gnu
+        run: rustup component add rust-src --toolchain nightly-2025-05-01-x86_64-unknown-linux-gnu
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  #v4.2.2
         with:
           persist-credentials: false
@@ -551,11 +551,11 @@ jobs:
   #   persist-credentials: false
   #     - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
   #       with:
-  #         toolchain: nightly-2025-06-16
+  #         toolchain: nightly-2025-05-01
   #         override: true
   #         components: rustfmt, clippy
   #     - name: Add rust-src
-  #       run: rustup component add rust-src --toolchain nightly-2025-06-16-x86_64-unknown-linux-gnu
+  #       run: rustup component add rust-src --toolchain nightly-2025-05-01-x86_64-unknown-linux-gnu
   #     - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  #v4.2.2
   #     - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
   #       with:
@@ -594,7 +594,7 @@ jobs:
 
       - uses: dtolnay/rust-toolchain@4f94fbe7e03939b0e674bcc9ca609a16088f63ff  #nightly branch, TODO: update when required
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -614,7 +614,7 @@ jobs:
   #   persist-credentials: false
   #     - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
   #       with:
-  #         toolchain: nightly-2025-06-16
+  #         toolchain: nightly-2025-05-01
   #         override: true
   #         components: rustfmt, clippy
   #     - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -640,7 +640,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -666,7 +666,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -696,7 +696,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -709,7 +709,7 @@ jobs:
   python-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     needs: [build, library-tests, docs]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
@@ -725,7 +725,7 @@ jobs:
           python-version: "3.12"
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - name: Install cmake
@@ -744,7 +744,7 @@ jobs:
   accuracy-measurement-tests:
     permissions:
       contents: read
-    runs-on: non-gpu,non-sgx
+    runs-on: [ non-gpu, non-sgx ]
     needs: [build, library-tests, docs, python-tests, python-integration-tests]
     env:
       EVM_VERIFIER_EZKL_TOKEN: ${{ secrets.EVM_VERIFIER_EZKL_TOKEN }}
@@ -760,7 +760,7 @@ jobs:
           python-version: "3.12"
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -798,7 +798,7 @@ jobs:
           python-version: "3.11"
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -858,7 +858,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - uses: baptiste0928/cargo-install@91c5da15570085bcde6f4d7aed98cb82d6769fd3  #v3.3.0
@@ -866,7 +866,7 @@ jobs:
           crate: cargo-nextest
           locked: true
       - name: Run ios tests
-        run: CARGO_BUILD_TARGET=aarch64-apple-darwin RUSTUP_TOOLCHAIN=nightly-2025-06-16-aarch64-apple-darwin cargo test --test ios_integration_tests --features ios-bindings-test --no-default-features
+        run: CARGO_BUILD_TARGET=aarch64-apple-darwin RUSTUP_TOOLCHAIN=nightly-2025-05-01-aarch64-apple-darwin cargo test --test ios_integration_tests --features ios-bindings-test --no-default-features
 
   swift-package-tests:
     permissions:
@@ -884,7 +884,7 @@ jobs:
 
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
       - name: Build EzklCoreBindings

.github/workflows/static-analysis.yml

@@ -17,7 +17,7 @@ jobs:
           persist-credentials: false
       - uses: actions-rust-lang/setup-rust-toolchain@fb51252c7ba57d633bc668f941da052e410add48  #v1.0.6
         with:
-          toolchain: nightly-2025-06-16
+          toolchain: nightly-2025-05-01
           override: true
           components: rustfmt, clippy
 

Cargo.toml

@@ -227,6 +227,7 @@ default = [
     "precompute-coset",
     "no-banner",
     "parallel-poly-read",
+    "reusable-verifier",
 ]
 onnx = ["dep:tract-onnx"]
 python-bindings = ["pyo3", "pyo3-log", "pyo3-async-runtimes", "pyo3-stub-gen"]

rust-toolchain

@@ -1,3 +1,3 @@
 [toolchain]
-channel = "nightly-2025-06-16"
+channel = "nightly-2025-05-01"
 components = ["rustfmt", "clippy"]

src/bindings/python.rs

@@ -142,6 +142,9 @@ struct PyRunArgs {
     #[pyo3(get, set)]
     /// int:  The denominator in the fixed point representation used when quantizing parameters
     pub param_scale: crate::Scale,
+    /// int: The scale to rebase to (optional). If None, we rebase to the max of input_scale and param_scale
+    /// This is an advanced parameter that should be used with caution
+    pub rebase_scale: Option<crate::Scale>,
     #[pyo3(get, set)]
     /// int: If the scale is ever > scale_rebase_multiplier * input_scale then the scale is rebased to input_scale (this a more advanced parameter, use with caution)
     pub scale_rebase_multiplier: u32,
@@ -208,6 +211,7 @@ impl From<PyRunArgs> for RunArgs {
             bounded_log_lookup: py_run_args.bounded_log_lookup,
             input_scale: py_run_args.input_scale,
             param_scale: py_run_args.param_scale,
+            rebase_scale: py_run_args.rebase_scale,
             num_inner_cols: py_run_args.num_inner_cols,
             scale_rebase_multiplier: py_run_args.scale_rebase_multiplier,
             lookup_range: py_run_args.lookup_range,
@@ -234,6 +238,7 @@ impl Into<PyRunArgs> for RunArgs {
             bounded_log_lookup: self.bounded_log_lookup,
             input_scale: self.input_scale,
             param_scale: self.param_scale,
+            rebase_scale: self.rebase_scale,
             num_inner_cols: self.num_inner_cols,
             scale_rebase_multiplier: self.scale_rebase_multiplier,
             lookup_range: self.lookup_range,

src/graph/node.rs

@@ -695,8 +695,8 @@ impl Node {
         opkind = opkind.homogenous_rescale(in_scales.clone())?.into();
         let mut out_scale = opkind.out_scale(in_scales.clone())?;
         // rescale the inputs if necessary to get consistent fixed points, we select the largest scale (highest precision)
-        let global_scale = scales.get_max();
-        opkind = RebaseScale::rebase(opkind, global_scale, out_scale, scales.rebase_multiplier);
+        let rebase_scale = scales.get_rebase_scale();
+        opkind = RebaseScale::rebase(opkind, rebase_scale, out_scale, scales.rebase_multiplier);
 
         out_scale = opkind.out_scale(in_scales)?;
 

src/graph/vars.rs

@@ -250,6 +250,8 @@ pub struct VarScales {
     pub params: crate::Scale,
     /// Multiplier for scale rebasing
     pub rebase_multiplier: u32,
+    /// rebase scale factor (optional). if None, we rebase to the max of input_scale and param_scale
+    pub rebase_scale: Option<crate::Scale>,
 }
 
 impl std::fmt::Display for VarScales {
@@ -269,11 +271,21 @@ impl VarScales {
         std::cmp::min(self.input, self.params)
     }
 
+    /// Returns the scale to rebase to, if specified
+    pub fn get_rebase_scale(&self) -> crate::Scale {
+        if let Some(rebase_scale) = self.rebase_scale {
+            rebase_scale
+        } else {
+            self.get_max()
+        }
+    }
+
     /// Creates VarScales from runtime arguments
     pub fn from_args(args: &RunArgs) -> Self {
         Self {
             input: args.input_scale,
             params: args.param_scale,
+            rebase_scale: args.rebase_scale,
             rebase_multiplier: args.scale_rebase_multiplier,
         }
     }

src/lib.rs

@@ -288,6 +288,10 @@ pub struct RunArgs {
     /// Higher values provide more precision but increase circuit complexity
     #[cfg_attr(all(feature = "ezkl", not(target_arch = "wasm32")), arg(long, default_value = "7", value_hint = clap::ValueHint::Other))]
     pub param_scale: Scale,
+    /// Scale to rebase to when the input scale exceeds rebase_scale * multiplier. If None we rebase to the max of input_scale and param_scale
+    /// This is an advanced parameter that should be used with caution
+    #[cfg_attr(all(feature = "ezkl", not(target_arch = "wasm32")), arg(long, required = false, value_hint = clap::ValueHint::Other))]
+    pub rebase_scale: Option<Scale>,
     /// Scale rebase threshold multiplier
     /// When scale exceeds input_scale * multiplier, it is rebased to input_scale
     /// Advanced parameter that should be used with caution
@@ -378,6 +382,7 @@ impl Default for RunArgs {
             bounded_log_lookup: false,
             input_scale: 7,
             param_scale: 7,
+            rebase_scale: None,
             scale_rebase_multiplier: 1,
             lookup_range: (-32768, 32768),
             logrows: 17,

src/tensor/ops.rs

@@ -329,7 +329,7 @@ pub fn resize<T: TensorType + Send + Sync>(
 
     let cartesian_coord: Vec<Vec<usize>> = new_shape
         .iter()
-        .map(|d| (0..*d))
+        .map(|d| 0..*d)
         .multi_cartesian_product()
         .collect();
 
@@ -1218,7 +1218,7 @@ pub fn intercalate_values<T: TensorType>(
     let cartesian_coord = output
         .dims()
         .iter()
-        .map(|d| (0..*d))
+        .map(|d| 0..*d)
         .multi_cartesian_product()
         .collect::<Vec<_>>();
 
@@ -1263,7 +1263,7 @@ pub fn one_hot(
     let cartesian_coord = output
         .dims()
         .iter()
-        .map(|d| (0..*d))
+        .map(|d| 0..*d)
         .multi_cartesian_product()
         .collect::<Vec<_>>();
 
@@ -1341,7 +1341,7 @@ pub fn pad<T: TensorType>(
     let cartesian_coord = image
         .dims()
         .iter()
-        .map(|d| (0..*d))
+        .map(|d| 0..*d)
         .multi_cartesian_product()
         .collect::<Vec<_>>();