Would definitely recommend the book "Staff Engineer" by Will Larson. Comments may not do justice to the full scope of a staff engineer, but generally in terms of how you want to be thinking: while your engineer skills have to be great, your main impact comes from influencing decisions, keeping the team on the right tracks, and being the one to ensure everyone's on the same page and has the necessary context to get their job done.