Simon Willis has stepped down from the post of Sri Lanka Cricket's high-performance manager, less than two years into a three-year contract.