Set strict=True in compare_mlx_and_py

Author: jessegrabowski

tests/link/mlx/test_basic.py

@@ -54,7 +54,7 @@ def compare_mlx_and_py(
 
     """
     if assert_fn is None:
-        assert_fn = partial(np.testing.assert_allclose, rtol=1e-4)
+        assert_fn = partial(np.testing.assert_allclose, rtol=1e-4, strict=True)
 
     if any(inp.owner is not None for inp in graph_inputs):
         raise ValueError("Inputs must be root variables")