mixed precision fix

Author: Jammy2211

autoarray/structures/arrays/kernel_2d.py

@@ -612,6 +612,7 @@ def convolved_image_from(
         image,
         blurring_image,
         jax_method="direct",
+        use_mixed_precision : bool = False,
         xp=np,
     ):
         """
@@ -742,6 +743,9 @@ def convolved_image_from(
                 new_shape=image_shape_original, mask_pad_value=0
             )
 
+        if use_mixed_precision:
+            blurred_image = blurred_image.astype(jnp.float32)
+
         return blurred_image
 
     def convolved_mapping_matrix_from(
@@ -1120,6 +1124,7 @@ def convolved_mapping_matrix_via_real_space_from(
             blurring_mask=blurring_mask,
             xp=xp,
         )
+
         # 6) Real-space convolution, broadcast kernel over source axis
         kernel = self.stored_native.array