warn, rather than err, on non-split output when a split was requested
warn, rather than err, on non-split output when a split was requested

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