augax._ks_dispatch

augax._ks_dispatch = <jax._src.custom_batching.custom_vmap object>[source]