For anyone interested: We solved this by building an OAuth integration to Snowflake ourselfs using Entra ID: 

https://community.snowflake.com/s/article/External-oAuth-Token-Generation-using-Azure-AD

We also created some simple Python and R-packages that includes methods to generate tokens and read/write data to Snowflake using spark and sparklyr for R.