I've been running GraphiteOS on my Pixel 4a (it is a backup phone) and obviously it doesn't have the battery update but I can attest that it works great with it.