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