I think the relevant part of the code is in this other repo:

https://github.com/twitter/the-algorithm

Not sure if it has what you were looking for (and maybe you already checked this repo, too!), but it's more relevant than the linked repo imo