Rishabh Pant was stumped after just four balls at the crease as Australia rallied to take three quick wickets against India.