stringjax_tools.auto_vectorise.set_auto_vmap_defaults

stringjax_tools.auto_vectorise.set_auto_vmap_defaults#

set_auto_vmap_defaults(**sample_ranks)#

Update global default sample ranks used by auto_vmap().

Return type:

dict[str, int]