they both should be equivalent, both in terms of numerical stability and computation. I wonder if there are some non-deterministic operations you are using, and that’s the cause of the difference in performance.