Normally Computer Science, certainly not Math. My university had a separate graduate program for Artificial Intelligence. Maybe also Neuroscience or Psychology

Normally Computer Science, certainly not Math. My university had a separate graduate program for Artificial Intelligence. Maybe also Neuroscience or Psychology, but they usually use other kinds of neural network models, which are more biologically plausible.