Cómo agregar un nuevo set de orígenes con gradle kotlin-dsl

Quiero agregar un set de orígenes src/gen/java . Con groovy esto es bastante fácil y ya se describe en https://discuss.gradle.org/t/how-to-use-gradle-with-generated-sources/9401/5

 sourceSets { gen { java.srcDir "src/gen/java" } } 

Pero me quedé con el kotlin-dsl para agregar uno nuevo. Todo lo que tengo es:

 java { sourceSets { } } 

¿Alguien puede ayudar aquí?