“A federal Medicaid rule, known as the IMD rule, prohibits reimbursement for mental healthcare provided in facilities with more than 16 beds.”