I’ve seen it mentioned in Sefarim. I just checked up Reb Reuven Margalis, and he brings this from the Sefer Vayakhel Moshe (entry 54), the Chida (Pesach Einayim on Avos Perek 5 on that Mishna of Machlokes L’shem Shamayim), and the Malbim (Torah Ohr Chukas 147)
By the way, there is no doubt about Mitzvos after Moshiach comes. The question is about after T’chiya.