Implement optimized stake distribution computation

When available, or fallback to the legacy computation

Author: jpraynaud