Unable to join session host to Entra ID

[ShakthiS16]

While creating the session host using ARM templates, am not able to join the Session host to Entra ID. The aad join property is set to true in the session host JSON after its creation. I even added the AADLoginForWindows extension to the session host. I understand that the aadjoin property is set to public preview, but isn't there any alternative to achieve this through ARM templates as there aren't any REST APIs available for session host creation.