Increase parallelism and timeouts for two more tests, lest they sporadically time out.
Increase parallelism and timeouts for two more tests, lest they sporadically time out.

PiperOrigin-RevId: 292958510

tensorflow/probability,tensorflow/probability