Machine Learning Research at Waymo — BranchFactor