optimize for GPU (avoid MLE error)