You can see all the supported attributes here:
Azure AD > Enterprise Applications > Salesforce > Mapping > Provision Azure Active Directory Users > Click on Add New Mapping link and under Source attribute drop down.
The attribute list for Azure Active Directory doesn't include group membership attribute. You can request additional attributes you would like to see supported here.
As of now, you can only use appRoles to provision users in Salesforce with specific roles.
-----------------------------------------------------------------------------------------------------------
Please "Accept the answer" if the information helped you. This will help us and others in the community as well.