diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix
index 1ad591e13be6..338711d80f97 100644
--- a/pkgs/development/python-modules/tensordict/default.nix
+++ b/pkgs/development/python-modules/tensordict/default.nix
@@ -70,13 +70,11 @@ buildPythonPackage rec {
       # +  where tensor(False) = <built-in method all of Tensor object at 0x7ffe49bf87d0>()
       "test_mp"
 
-      # torch._dynamo.exc.BackendCompilerFailed
-      # Requires a more recent version of triton
-      # Re-enable when https://github.com/NixOS/nixpkgs/pull/328247 is merged
+      # torch._dynamo.exc.InternalTorchDynamoError: RuntimeError: to_module requires TORCHDYNAMO_INLINE_INBUILT_NN_MODULES to be set.
       "test_functional"
-      "test_linear"
-      "test_seq"
-      "test_seq_lmbda"
+
+      # hangs forever on some CPUs
+      "test_map_iter_interrupt_early"
     ]
     ++ lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [
       # RuntimeError: internal error
@@ -86,9 +84,6 @@ buildPythonPackage rec {
 
       # _queue.Empty errors in multiprocessing tests
       "test_isend"
-
-      # hangs forever
-      "test_map_iter_interrupt_early"
     ];
 
   disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [
diff --git a/pkgs/development/python-modules/torchrl/default.nix b/pkgs/development/python-modules/torchrl/default.nix
index 990e4df76be4..a1555b29357e 100644
--- a/pkgs/development/python-modules/torchrl/default.nix
+++ b/pkgs/development/python-modules/torchrl/default.nix
@@ -167,6 +167,10 @@ buildPythonPackage rec {
     # assert torch.get_num_threads() == max(1, init_threads - 3)
     # AssertionError: assert 23 == 21
     "test_auto_num_threads"
+
+    # Flaky (hangs indefinitely on some CPUs)
+    "test_gae_multidim"
+    "test_gae_param_as_tensor"
   ];
 
   meta = {