Change Travis configuration to use pytest-xdist.
Change Travis configuration to use pytest-xdist.

Since the Travis VM is dual core, run two tests in parallel.

tensorflow/probability,google/jax,google/jax,tensorflow/probability,google/jax,google/jax