I've noticed my network has to change to 2G from 4G in order to send or receive sms. So I think Airtel's implementation of IMS stack is restricted only to voice and not sms. IMS stack is what enables devices to use volte, vilte, vowifi, etc. So SMS still relies on 2G to get around. Also authentication with Airtel's IMS core occurs over IP. It is upto the telecom carrier how this authentication process happens. Maybe Airtel had implemented the use of regular Airtel 4G for primary authentication for IMS, so only after that happens will vowifi and volte will work? Mind you, you can switch off data services but IMS stack of your phone can still use your phones data to connect to its server.
Jio on the other hand is using all IMS core. They obviously can't afford not to cos they don't have any other network than IP-only 4G. So maybe their authentication can happen without needing regular 4G to be available. This probably is the reason for being able to use all IMS services despite being on wifi.