Using Michigan as an example, maybe you have people vote for 5 people from the party of their choosing. Doesn't need to be ranked choice as you've already declared the party you want representing you. Once the votes are tallied and the state knows the split, they take the top vote getters to fill the seats.
Obviously this is just a silly idea I came up with, but the more I think about it, the more I like it. I've already mentioned how this would give every person in the state the representation they want, but there's also another benefit. Lets say you have an issue or concern and need to reach out to a Rep. You have the option of contacting the Rep you think can best help you or who will be most receptive to your concerns.