added further models with BMI

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate.py

@@ -891,17 +891,19 @@ def apply_fertilizer_kernel(state):
         vs.Nmin_in = update(vs.Nmin_in, at[2:-2, 2:-2], npx.where((vs.doy_fert3[2:-2, 2:-2] == vs.DOY[vs.itt]), (vs.N_fert3[2:-2, 2:-2] * settings.dx * settings.dy * 100), vs.Nmin_in[2:-2, 2:-2]))
         vs.Nmin_in = update(vs.Nmin_in, at[2:-2, 2:-2], npx.where(vs.Nmin_in[2:-2, 2:-2] < 0, 0, vs.Nmin_in[2:-2, 2:-2]))
 
-        inf = vs.inf_mat_rz[2:-2, 2:-2] + vs.inf_pf_rz[2:-2, 2:-2] + vs.inf_pf_ss[2:-2, 2:-2]
+        inf = allocate(state.dimensions, ("x", "y"))
+        inf = update(inf, at[2:-2, 2:-2], vs.inf_mat_rz[2:-2, 2:-2] + vs.inf_pf_rz[2:-2, 2:-2] + vs.inf_pf_ss[2:-2, 2:-2])
         vs.inf_in_tracer = update(vs.inf_in_tracer, at[2:-2, 2:-2], npx.where((vs.doy_dist[2:-2, 2:-2] == vs.doy_fert1[2:-2, 2:-2]) | (vs.doy_dist[2:-2, 2:-2] == vs.doy_fert2[2:-2, 2:-2]) | (vs.doy_dist[2:-2, 2:-2] == vs.doy_fert3[2:-2, 2:-2]), 0, vs.inf_in_tracer[2:-2, 2:-2]))
-        vs.inf_in_tracer = update_add(vs.inf_in_tracer, at[2:-2, 2:-2], inf)
-        inf_ratio = npx.where((inf/settings.cum_inf_for_N_input) < 1, inf/settings.cum_inf_for_N_input, 1)
+        vs.inf_in_tracer = update_add(vs.inf_in_tracer, at[2:-2, 2:-2], inf[2:-2, 2:-2])
+        inf_ratio = allocate(state.dimensions, ("x", "y"))
+        inf_ratio = update(inf_ratio, at[2:-2, 2:-2], npx.where((inf[2:-2, 2:-2]/settings.cum_inf_for_N_input) < 1, inf[2:-2, 2:-2]/settings.cum_inf_for_N_input, 1))
         # dissolved nitrogen input as nitrate
-        vs.M_in = update(vs.M_in, at[2:-2, 2:-2], npx.where(vs.inf_in_tracer[2:-2, 2:-2] > 0, vs.Nmin_in[2:-2, 2:-2] * inf_ratio * _c1, 0))
+        vs.M_in = update(vs.M_in, at[2:-2, 2:-2], npx.where(vs.inf_in_tracer[2:-2, 2:-2] > 0, vs.Nmin_in[2:-2, 2:-2] * inf_ratio[2:-2, 2:-2] * _c1, 0))
         vs.ndep_s = update(vs.ndep_s, at[2:-2, 2:-2], 0)
         # wet nitrate deposition
-        vs.M_in = update_add(vs.M_in, at[2:-2, 2:-2], npx.where(inf > 0, vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5, 0))
-        vs.ndep_s = update_add(vs.ndep_s, at[2:-2, 2:-2], npx.where(inf > 0, vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5, 0))  
-        vs.C_in = update(vs.C_in, at[2:-2, 2:-2], npx.where(vs.inf_in_tracer[2:-2, 2:-2] > 0, vs.M_in[2:-2, 2:-2]/inf, 0))
+        vs.M_in = update_add(vs.M_in, at[2:-2, 2:-2], npx.where(inf[2:-2, 2:-2] > 0, vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5, 0))
+        vs.ndep_s = update_add(vs.ndep_s, at[2:-2, 2:-2], npx.where(inf[2:-2, 2:-2] > 0, vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5, 0))  
+        vs.C_in = update(vs.C_in, at[2:-2, 2:-2], npx.where(vs.inf_in_tracer[2:-2, 2:-2] > 0, vs.M_in[2:-2, 2:-2]/inf[2:-2, 2:-2], 0))
         # undissolved nitrogen input as ammonium
         vs.Nmin_rz = update_add(
             vs.Nmin_rz,
@@ -915,7 +917,7 @@ def apply_fertilizer_kernel(state):
             vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5,
         )
         vs.ndep_s = update_add(vs.ndep_s, at[2:-2, 2:-2], vs.kdep[2:-2, 2:-2] * settings.dx * settings.dy * (100/365) * 0.5)  
-        vs.Nmin_in = update_add(vs.Nmin_in, at[2:-2, 2:-2], -vs.Nmin_in[2:-2, 2:-2] * inf_ratio)
+        vs.Nmin_in = update_add(vs.Nmin_in, at[2:-2, 2:-2], -vs.Nmin_in[2:-2, 2:-2] * inf_ratio[2:-2, 2:-2])
         vs.Nmin_in = update(vs.Nmin_in, at[2:-2, 2:-2], npx.where((vs.Nmin_in[2:-2, 2:-2] < 0), 0, vs.Nmin_in[2:-2, 2:-2]))
         vs.inf_in_tracer = update(vs.inf_in_tracer, at[2:-2, 2:-2], npx.where((vs.inf_in_tracer[2:-2, 2:-2] > settings.cum_inf_for_N_input), 0, vs.inf_in_tracer[2:-2, 2:-2]))
 

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_medium_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_medium_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_medium_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_medium_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-rape_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-rape_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-rape_medium_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-rape_yellow-mustard_high_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

examples/plot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-rape_yellow-mustard_low_Nfert_slurm.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=72:00:00
+#SBATCH --time=32:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-rape_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate