chore: Set `OMP_NUM_THREADS` and related env vars prior to importing `numpy`

[ssheng]

OMP_NUM_THREADS must be set before numpy is imported for it to work, our current implementation doesn’t guarantee that.