Undergrad in Need of Grad School Advice? • r/artificial

I am currently going into my 2nd semester of my junior year, majoring in mathematical science. Last semester I took numerical linear algebra, and while difficult, I found it very interesting. We didn't do any machine learning but did talk about it a lot and I got to know python quite well. Since this class I have been juggling the idea of possibly going to grad school for machine learning/AI. I need some advice on what I should do.