I'm tried both the apk from the Protonmail website as well as the one from Aurora Store.
Notifications are switched on within the app and are enabled in Grapheneos. However, I'm not getting any notifications when new mail arrives.
Notifications work for other apps. From my understanding, the Proton Mail app doesn't need Google Framework to allow for notifications so I'm not sure what I'm missing.