я бы с радостью, но в бабеле за это отвечает отдельная категория плагинов, которые должны интегрироваться в @babel/core, то есть без контрибьюта в ядро бабеля такие преколы не реализовать
собственно return абы где уже сейчас можно кидать если включена опция парсера top level return. используется например в том же репле эффектора, код в котором оборачивается в функцию после компиляции, а значит бабель должен нормально отнестись к return в корне