I recently started graduate school in mathematics, and I have been having some difficulty focusing on studying.

Difficulties focusing on studying math are common. There's an issue of many existing expositions being markedly suboptimal with respect to engaging the student.

I'm in my fourth year of math graduate school and I would be happy to offer suggestions specific to your situation. Feel free to PM me with your email address if you'd like to correspond.