Merge "Increase memory for distribution SFT jobs"