Test for matmul is flaky: sometimes the difference is NaN: https://github.com/microsoft/onnx-script/actions/runs/4470190371/jobs/7853362946#step:5:3924 This starts to happen when we use PyTorch 2.0