This is interesting - I wonder how hard it would be to do this with a transformer model.