diff --git a/extra_tests/regression_fits/central_16.json b/extra_tests/regression_fits/central_16.json
index 91f6b14d1d..8322d13ff2 100644
--- a/extra_tests/regression_fits/central_16.json
+++ b/extra_tests/regression_fits/central_16.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 19.067643642425537,
+      "Total": 19.351093769073486,
       "start": 0.0,
-      "replica_set": 0.2691524028778076,
-      "replica_fitted": 19.067547082901,
-      "replica_set_to_replica_fitted": 18.798394680023193
+      "replica_set": 0.25216174125671387,
+      "replica_fitted": 19.35101819038391,
+      "replica_set_to_replica_fitted": 19.098856449127197
     },
     "cputime": {
-      "Total": 20.805063351,
+      "Total": 21.418973193,
       "start": 0.0,
-      "replica_set": 0.5839991970000007,
-      "replica_fitted": 20.804965238,
-      "replica_set_to_replica_fitted": 20.220966041
+      "replica_set": 0.56906414,
+      "replica_fitted": 21.418895488,
+      "replica_set_to_replica_fitted": 20.849831348
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/diagonal_45.json b/extra_tests/regression_fits/diagonal_45.json
index dacb427991..b435886640 100644
--- a/extra_tests/regression_fits/diagonal_45.json
+++ b/extra_tests/regression_fits/diagonal_45.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 17.70769500732422,
+      "Total": 17.985970497131348,
       "start": 0.0,
-      "replica_set": 0.25739622116088867,
-      "replica_fitted": 17.70762538909912,
-      "replica_set_to_replica_fitted": 17.450229167938232
+      "replica_set": 0.27022218704223633,
+      "replica_fitted": 17.98584008216858,
+      "replica_set_to_replica_fitted": 17.715617895126343
     },
     "cputime": {
-      "Total": 19.413020977000002,
+      "Total": 19.905585440000003,
       "start": 0.0,
-      "replica_set": 0.5726430499999999,
-      "replica_fitted": 19.412949704,
-      "replica_set_to_replica_fitted": 18.840306654000003
+      "replica_set": 0.5847023880000002,
+      "replica_fitted": 19.905453233000003,
+      "replica_set_to_replica_fitted": 19.320750845
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/feature_scaling_81.json b/extra_tests/regression_fits/feature_scaling_81.json
index a58c1771ae..b97434b685 100644
--- a/extra_tests/regression_fits/feature_scaling_81.json
+++ b/extra_tests/regression_fits/feature_scaling_81.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 19.08158278465271,
+      "Total": 18.170032501220703,
       "start": 0.0,
-      "replica_set": 0.39364171028137207,
-      "replica_fitted": 19.081433534622192,
-      "replica_set_to_replica_fitted": 18.68779182434082
+      "replica_set": 0.27858686447143555,
+      "replica_fitted": 18.16972517967224,
+      "replica_set_to_replica_fitted": 17.891138315200806
     },
     "cputime": {
-      "Total": 20.744764767,
+      "Total": 19.786388953,
       "start": 0.0,
-      "replica_set": 0.7075304789999999,
-      "replica_fitted": 20.744614046000002,
-      "replica_set_to_replica_fitted": 20.037083567
+      "replica_set": 0.5939304989999998,
+      "replica_fitted": 19.786028391,
+      "replica_set_to_replica_fitted": 19.192097892
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/flavour_29.json b/extra_tests/regression_fits/flavour_29.json
index 2200544f1b..ce759a4124 100644
--- a/extra_tests/regression_fits/flavour_29.json
+++ b/extra_tests/regression_fits/flavour_29.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 7.596189260482788,
+      "Total": 7.384192943572998,
       "start": 0.0,
-      "replica_set": 0.43732333183288574,
-      "replica_fitted": 7.596093416213989,
-      "replica_set_to_replica_fitted": 7.1587700843811035
+      "replica_set": 0.29032063484191895,
+      "replica_fitted": 7.384095668792725,
+      "replica_set_to_replica_fitted": 7.093775033950806
     },
     "cputime": {
-      "Total": 8.141307009,
+      "Total": 7.951215819,
       "start": 0.0,
-      "replica_set": 0.751879143,
-      "replica_fitted": 8.141209577,
-      "replica_set_to_replica_fitted": 7.389330434
+      "replica_set": 0.6030387780000002,
+      "replica_fitted": 7.951116704,
+      "replica_set_to_replica_fitted": 7.348077926
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/no_csr_613.json b/extra_tests/regression_fits/no_csr_613.json
index edbf1f7904..6ee8a669ef 100644
--- a/extra_tests/regression_fits/no_csr_613.json
+++ b/extra_tests/regression_fits/no_csr_613.json
@@ -77,24 +77,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 15.619823932647705,
+      "Total": 15.619742631912231,
       "start": 0.0,
-      "replica_set": 0.4021031856536865,
-      "replica_fitted": 15.619724035263062,
-      "replica_set_to_replica_fitted": 15.217620849609375
+      "replica_set": 0.2965066432952881,
+      "replica_fitted": 15.619616270065308,
+      "replica_set_to_replica_fitted": 15.32310962677002
     },
     "cputime": {
-      "Total": 16.929444454,
+      "Total": 16.903149643,
       "start": 0.0,
-      "replica_set": 0.7151704030000001,
-      "replica_fitted": 16.929343175,
-      "replica_set_to_replica_fitted": 16.214172771999998
+      "replica_set": 0.6058529689999999,
+      "replica_fitted": 16.903021783999996,
+      "replica_set_to_replica_fitted": 16.297168815
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/no_lagrange_27.json b/extra_tests/regression_fits/no_lagrange_27.json
index 768020e206..988d5e9b74 100644
--- a/extra_tests/regression_fits/no_lagrange_27.json
+++ b/extra_tests/regression_fits/no_lagrange_27.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 22.64839267730713,
+      "Total": 22.71401810646057,
       "start": 0.0,
-      "replica_set": 0.2564716339111328,
-      "replica_fitted": 22.64828395843506,
-      "replica_set_to_replica_fitted": 22.391812324523926
+      "replica_set": 0.2537527084350586,
+      "replica_fitted": 22.713902711868286,
+      "replica_set_to_replica_fitted": 22.460150003433228
     },
     "cputime": {
-      "Total": 24.290361283,
+      "Total": 24.164911617,
       "start": 0.0,
-      "replica_set": 0.5724489049999999,
-      "replica_fitted": 24.290250286,
-      "replica_set_to_replica_fitted": 23.717801381
+      "replica_set": 0.5701306190000004,
+      "replica_fitted": 24.164794980000003,
+      "replica_set_to_replica_fitted": 23.594664361
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/no_msr_92.json b/extra_tests/regression_fits/no_msr_92.json
index 598d53d971..77236b6bb5 100644
--- a/extra_tests/regression_fits/no_msr_92.json
+++ b/extra_tests/regression_fits/no_msr_92.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 32.641329288482666,
+      "Total": 31.41426372528076,
       "start": 0.0,
-      "replica_set": 0.2623887062072754,
-      "replica_fitted": 32.641178607940674,
-      "replica_set_to_replica_fitted": 32.3787899017334
+      "replica_set": 0.25605225563049316,
+      "replica_fitted": 31.41413378715515,
+      "replica_set_to_replica_fitted": 31.158081531524658
     },
     "cputime": {
-      "Total": 34.76780325,
+      "Total": 33.432995651,
       "start": 0.0,
-      "replica_set": 0.5775030430000001,
-      "replica_fitted": 34.767650695,
-      "replica_set_to_replica_fitted": 34.190147652
+      "replica_set": 0.5719856840000004,
+      "replica_fitted": 33.432864246,
+      "replica_set_to_replica_fitted": 32.860878562
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/no_sumrules_18.json b/extra_tests/regression_fits/no_sumrules_18.json
index aec00f2724..587b3bd8bb 100644
--- a/extra_tests/regression_fits/no_sumrules_18.json
+++ b/extra_tests/regression_fits/no_sumrules_18.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 27.75906491279602,
+      "Total": 26.803892612457275,
       "start": 0.0,
-      "replica_set": 0.31566333770751953,
-      "replica_fitted": 27.758901834487915,
-      "replica_set_to_replica_fitted": 27.443238496780396
+      "replica_set": 0.2616863250732422,
+      "replica_fitted": 26.803722143173218,
+      "replica_set_to_replica_fitted": 26.542035818099976
     },
     "cputime": {
-      "Total": 29.705988494,
+      "Total": 28.404558843000004,
       "start": 0.0,
-      "replica_set": 0.6208636369999994,
-      "replica_fitted": 29.705823335999998,
-      "replica_set_to_replica_fitted": 29.084959699
+      "replica_set": 0.5790414630000003,
+      "replica_fitted": 28.404386691,
+      "replica_set_to_replica_fitted": 27.825345228
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/no_vsr_54.json b/extra_tests/regression_fits/no_vsr_54.json
index 0061d5c84b..f743e53f06 100644
--- a/extra_tests/regression_fits/no_vsr_54.json
+++ b/extra_tests/regression_fits/no_vsr_54.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 18.800585746765137,
+      "Total": 18.946770191192627,
       "start": 0.0,
-      "replica_set": 0.26001858711242676,
-      "replica_fitted": 18.800503730773926,
-      "replica_set_to_replica_fitted": 18.5404851436615
+      "replica_set": 0.2613098621368408,
+      "replica_fitted": 18.946637868881226,
+      "replica_set_to_replica_fitted": 18.685328006744385
     },
     "cputime": {
-      "Total": 20.795346321,
+      "Total": 20.809097111,
       "start": 0.0,
-      "replica_set": 0.5764106189999998,
-      "replica_fitted": 20.795262424,
-      "replica_set_to_replica_fitted": 20.218851805
+      "replica_set": 0.5740524349999996,
+      "replica_fitted": 20.808963461,
+      "replica_set_to_replica_fitted": 20.234911026
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/normal_fit_72.json b/extra_tests/regression_fits/normal_fit_72.json
index d1525f11bd..0a1144ce59 100644
--- a/extra_tests/regression_fits/normal_fit_72.json
+++ b/extra_tests/regression_fits/normal_fit_72.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 30.5302574634552,
+      "Total": 31.17136573791504,
       "start": 0.0,
-      "replica_set": 0.266298770904541,
-      "replica_fitted": 30.530073642730713,
-      "replica_set_to_replica_fitted": 30.263774871826172
+      "replica_set": 0.29683351516723633,
+      "replica_fitted": 31.171199083328247,
+      "replica_set_to_replica_fitted": 30.87436556816101
     },
     "cputime": {
-      "Total": 33.634691678,
+      "Total": 33.843777033,
       "start": 0.0,
-      "replica_set": 0.5827569669999999,
-      "replica_fitted": 33.634505951,
-      "replica_set_to_replica_fitted": 33.051748984
+      "replica_set": 0.6099015160000008,
+      "replica_fitted": 33.843608378999996,
+      "replica_set_to_replica_fitted": 33.233706862999995
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }
diff --git a/extra_tests/regression_fits/trainable_prepro_61.json b/extra_tests/regression_fits/trainable_prepro_61.json
index b920b71374..c0a68b1ca9 100644
--- a/extra_tests/regression_fits/trainable_prepro_61.json
+++ b/extra_tests/regression_fits/trainable_prepro_61.json
@@ -71,24 +71,24 @@
   ],
   "timing": {
     "walltime": {
-      "Total": 19.335994243621826,
+      "Total": 19.938401699066162,
       "start": 0.0,
-      "replica_set": 0.2786440849304199,
-      "replica_fitted": 19.3359375,
-      "replica_set_to_replica_fitted": 19.05729341506958
+      "replica_set": 0.2674376964569092,
+      "replica_fitted": 19.938326358795166,
+      "replica_set_to_replica_fitted": 19.670888662338257
     },
     "cputime": {
-      "Total": 21.199827569999997,
+      "Total": 21.699957106,
       "start": 0.0,
-      "replica_set": 0.5940154189999998,
-      "replica_fitted": 21.199769150999998,
-      "replica_set_to_replica_fitted": 20.605753732
+      "replica_set": 0.5827538259999994,
+      "replica_fitted": 21.699880133,
+      "replica_set_to_replica_fitted": 21.117126307
     }
   },
   "version": {
     "tensorflow": "2.15.0, mkl=??",
     "numpy": "1.26.4",
-    "nnpdf": "4.0.9.post120.dev0+9cfa7c2df",
-    "validphys": "4.0.9.post120.dev0+9cfa7c2df"
+    "nnpdf": "4.0.9.post121.dev0+306ca7c6e",
+    "validphys": "4.0.9.post121.dev0+306ca7c6e"
   }
 }