This version runs fast on Nvidia cluster