There are no restrictions on travel to and from Mexico for medical care for US citizens.
For other countries, check if there are requirements for your return home.