For anyone who has a similar issue like me, it turned out the nvme ssd i was using didn't agree with the notebook when it came to sleep states. I swapped it with an WD Blue SN570 and now everything works as expected.