Renamed main function to autovmap to be more consistent with package name.
Old name auto_vmap still available as alias for compatibility.
Fixed issue where pytree flattening was unnecessarily applied even when argument specification was None.
Renamed main function to autovmap to be more consistent with package name.
Old name auto_vmap still available as alias for compatibility.
Fixed issue where pytree flattening was unnecessarily applied even when argument specification was None.