jaxvacua.util.get_auto_vmap_default_shapes# get_auto_vmap_default_shapes()# Return a copy of the active JAXVacua-local auto-vmap shapes. Return type: dict[str, Any]
