stringjax_tools.auto_vectorise.auto_vmap_defaults

stringjax_tools.auto_vectorise.auto_vmap_defaults#

auto_vmap_defaults(**sample_ranks)#

Temporarily update global sample-rank defaults inside a with block.

Return type:

Iterator[dict[str, int]]