To pick up the story after the events of Demon Slayer Infinity Castle, you'll want to begin with chapter 158 of Koyoharu Gotouge's manga. If you're collecting in trades, it's part of the end of Volume ...