diff --git a/tests/python/test_ad_ndarray.py b/tests/python/test_ad_ndarray.py
index 81a13a37a8..a48b1ba340 100644
--- a/tests/python/test_ad_ndarray.py
+++ b/tests/python/test_ad_ndarray.py
@@ -10,6 +10,9 @@
 if has_pytorch():
     import torch
 
+
+pytestmark = pytest.mark.flaky(retries=5)
+
 archs_support_ndarray_ad = [ti.cpu, ti.cuda]
 
 
diff --git a/tests/python/test_sparse_matrix.py b/tests/python/test_sparse_matrix.py
index f318372921..c2b9cc8903 100644
--- a/tests/python/test_sparse_matrix.py
+++ b/tests/python/test_sparse_matrix.py
@@ -3,6 +3,8 @@
 import taichi as ti
 from tests import test_utils
 
+pytestmark = pytest.mark.flaky(retries=5)
+
 
 @pytest.mark.parametrize(
     "dtype, storage_format",