Implement Non-negative matrix factorisation from the Alpha-Zero paper in (ideally HF) language models

[sdtblck]

Alpha zero paper: https://arxiv.org/pdf/2111.09259.pdf

NMF technique is detailed in section 7.1