augax._ks_sample_batched¶ augax._ks_sample_batched = <jax._src.custom_batching.custom_vmap object>[source]¶ u is (B, ...), seed is (B,). Type: One FFI call for a batch