Training and inference code are from
https://google-research.github.io/self-organising-systems/isonca/

Current model is: