According to Article III of the U.S. Constitution, the

legislative branch has the power to create laws that determine punishment for federal crimes
legislative branch has the power to conduct trials when the federal courts are not in session
judicial branch has the power to create laws that determine punishment for federal crimes
judicial branch has the power to conduct trials when the federal courts are not in session