I've got ADHD, working remotely from home is an infinity% improvement over the open office I was working in before.
I'd say it's about equal to, or maybe a slight improvement over, the the "team room" I was in before the open office where it was me and ~10 other engineers all working on generally the same categorical area of a product (in my case the hardware team).
The only thing that I could see possibly being an improvement would be an even smaller, very quiet room that is located within walking distance of my home where only my direct sub-team mates and I sit, they're all people I like, and we all have good "when it's okay to talk" etiquette.
I feel like that would be an incredibly hard thing to arrange. So, fully remote, working from home, seems to me to be the next best option.