[Sigmas] Keep sigmas on CPU (#6173)

Commit

1 year ago

[Sigmas] Keep sigmas on CPU (#6173) * correct * Apply suggestions from code review * make style