In general

A State agency may provide transitional supplemental nutrition assistance program benefits—

(A) to a household that ceases to receive cash assistance under a State program funded under part A of title IV of the Social Security Act ( 42 U.S.C. 601 et seq.); or

(B) at the option of the State, to a household with children that ceases to receive cash assistance under a State-funded public assistance program.