...

android {

    ...

    dexOptions {
           dexInProcess true
           // maxProcessCount 4 -> Default is 4
           javaMaxHeapSize "2G"
    }

    ...
}
...