Quick follow-up on your question. I'm happy to hear you are considering improving Ristretto. Even if you are just interested in improving the code for yourself, without pushing anything to Github, I'm happy to give you some pointers. This
Ristretto page, chapter "Limitations", mentions 2 points how Ristretto can be improved.