Shankar Balachandran, IIT Madras