I would try and get the GSR axles just for consistency but really any B-series axle will work...

I had LS axles when I got my GSR tranny, and used them fine for years...