Some matrix field broadcasting results in `Failed to compile PTX code (ptxas received signal 9)` errors

[charleskawczynski]

For example: https://buildkite.com/clima/climacore-ci/builds/4458#01924384-ee2a-4328-a106-c6501a46657e.

Per Tim Besard on slack:

signal 9 comes from outside; OOM killer maybe?

I confirmed with him, this is OOM during compilation.

I'm not sure how we can fix this. I think we'll need to profile some simpler cases to see where the compiler is mostly allocating.