Where was Muhammad al-Baqir(as) during the Battle of Karbala and what followed?

During and in the aftermath of the Battle of Karbala where was Muhammad al-Baqir (as)? And under whose care? Seeing from his date of birth and the year in which the Battle of Karbala took place (source: Wikipedia) he was 3-4 years old at the time.