Public schools are not the US government. They're usually city or county.

There are many federal laws and policies enforced in the schools. The police arresting kids on school grounds are also usually state or local.