Given Mozilla's AI stance I can't see it as an improvement. I never want to side with NOT-Mozilla but the way they handled the Firefox AI stuff was messy at the bare minimum.
As a direct Proton Mail replacement I'm migrating to tuta.com. Which funnily enough just got a Thunderbird add-on recently.