[JAX] Remove GSPMD tests + adding guards and warning msg for GSPMD rules#2702

phu0ngng merged 2 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

phu0ngng:rm_gspmdphu0ngng/TransformerEngine:rm_gspmdCopy head branch name to clipboard