Just install your favorite desktop + mobile mail apps and you're fine.
If that can't be done with Protonmail, and you want to move your email out of the US, suggest FastMail, based in Australia.
Nothing really support IMAP search.
Every mua sucks at large mbox file.
-- Edit: I am a happy fastmail user
But I guess you mean search w/o local caching