Have some performance issues to work out still, but the TextKit2 editor update for Drafts should be ready soon, for those folks wanting the “full" Writing Tools experience.
I’m likely going ship at least one version with the editor defaulting to TextKit1 still and the new editor as an advanced option.