Remove dtype warning for `np.quantile` (#3188)
Remove dtype warning for `np.quantile` (#3188)

* drop the warning in index_to_gather

* fix dtype issue at quantile

* revert the change, the issue seems to be fixed
google/jax,tensorflow/probability,google/jax,google/jax,google/jax,tensorflow/probability