jaxvacua.util.set_auto_vmap_defaults

set_auto_vmap_defaults(**sample_ranks)

Update JAXVacua-local default sample ranks for auto_vmap().

These defaults are independent of stringjax_tools global defaults; importing JAXVacua therefore does not mutate behaviour in other packages. Changes affect subsequently decorated functions, while built-in JAXVacua methods use the defaults active when their classes were imported.

Return type:

dict[str, int]