jaxvacua.util.get_auto_vmap_defaults

jaxvacua.util.get_auto_vmap_defaults#

get_auto_vmap_defaults()#

Return a copy of the active JAXVacua-local auto-vmap ranks.

Return type:

dict[str, int]