If the rocks are about 1.5-2inches that should be fine. Only thing is, as with gravel, a lot of waste builds up underneath and in between; and even with a gravel cleaner, a lot of gunk builds up.
Make sure they give you the figures/no.s of the each test result, tell them to write it down beside each one (it can be very easy to muddle the nitrAte and nitrIte readings up, if you're not sure what they're about0.
Also, if the petshop says your figures seem a bit extreme or high; do not be tempted, on their advice to buy something to fix the nitrate, lock the ammonia or raise and lower your pH. These invariably do nothing and just causes more problems for your tank as they are temporary fixes only.
Axies can live in pH levels of 6.8-8.0
If ammonia or nitrIte are above 0 the best way to fix it, without chemicals, is to do frequent daily 30% partial only waterchanges. This will bring the toxins down so its safe enough for your axies to live in.
You could just leave it bare, it is by far the easiest thing for cleaning and feeding (I've only realised after looking after both my sisters babies and my axie larvae how easy it is). Almost tempting to remove the sand from some of our tanks but hubby would have a fit over it.