Updates to the logits estimator

Merged Amir MOHAMMADI requested to merge logits into master

Merge request reports