My advice: Take it easy - It seems that recovery time can vary greatly from person to person. I'm 6 months in and nowhere near functional yet. I plan that if/when I get better to take it easy for at least another 4 months to avoid a relapse. Then I would very slowly ease back into a full time schedule.

Grad school will always be there but you only get one brain.

that just my 5 cents. The decision of course lies with you

Best of Luck!
Happy to hear you "made it out of the woods"