Add references to paper definitions/lemmas

[fizruk]

Many Lean definitions we have correspond to some definition/lemma/theorem in the KS22 paper, so we should reference it properly, e.g.:

-- [KS22, Lemma 2.3]