Skip to content
Portrait of Arpit Dwivedi

Arpit Dwivedi

MS AeroAstro @ Stanford • Researcher @ ASL

Controls & Planning × Reinforcement Learning Robotics

About

I'm a master's student in Aeronautics & Astronautics at Stanford University, currently conducting research in the Autonomous Systems Lab advised by Prof. Marco Pavone. I am also working as a Research Assistant with Prof. Tim De Silva, using Reinforcement learning to study and model human decision-making in macroeconomics and behavioral economics. Before Stanford, I completed a B.Tech. in Mechanical Engineering with Honors and a minor in AI & Data Science at IIT Bombay, where I worked with Prof. Dwaipayan Mukherjee in the Controls & Computing Lab. My interests span the intersection of control theory, machine learning, and decision making - with a focus on autonomous driving and space vehicles.

Stanford University logo
M.S. in AeroAstro (2024–2026)
Stanford University
Rivian logo
Software Eng. Intern — Autonomy (2025)
Rivian Automotive
Autonomous Systems Lab logo
Graduate Researcher (2024–present)
Stanford ASL
IIT Bombay logo
B.Tech. Mechanical Eng. (2020–2024)
IIT Bombay
UBC logo
Research Intern (2023)
UBC, Vancouver
UMIC SeDriCa logo
Controls Engineer (2022–2024)
UMIC — SeDriCa

Education

M.Sc. in Aeronautics & Astronautics
Department of Aeronautics & Astronautics, Stanford University
2024–2026 · Stanford, CA, USA

Teaching Assistantships

  • ME 210: Introduction to Mechatronics (2026)
  • CME 215: Machine Learning and the Physical Sciences (2025)
B.Tech. in Mechanical Engineering (Honours)
Minor in Artificial Intelligence & Data Science
Indian Institute of Technology Bombay
2020–2024 · Mumbai, MH, India

Teaching Assistantships

  • CE 102: Engineering Mechanics (2022)

Professional Experience

Software Engineering Intern — Autonomy, Rivian Automotive
Summer 2025

Developing and testing autonomy stack modules for Rivian's vehicles, focusing on perception-planning integration and robust performance for real-world deployment.

Research Experience

Semantic Trajectory Generation for Goal-Oriented Spacecraft Rendezvous
Finalist for Best Paper Award
Yuji Takubo, Arpit Dwivedi, Sukeerth Ramkumar, Luis A. Pabon, Daniele Gammelli, Marco Pavone, Simone D'Amico • AIAA SCITECH, Jan 2026

This paper introduces SAGES (Semantic Autonomous Guidance Engine for Space), a trajectory-generation framework that translates natural-language commands into spacecraft trajectories that reflect high-level intent while respecting nonconvex constraints.

Paper · Website

FreeFlyer Solution
Imitation Learning for Trajectory Optimization

My work benchmarks sequence models - including Decision Transformer, S6-Mamba, and S4D - to warm-start nonlinear trajectory optimizers, with safety enforced through Sequential Convex Programming. The Decision Transformer demonstrated the strongest performance, achieving an 80% faster warm-start (0.37s), 42% lower runtime, and 14% higher feasibility compared to numerical solvers. I further deployed the learned controller on a Jetson AGX Orin within a ROS2 stack, where it achieved real-time open-loop execution and closed-loop flight through integrated MPC on Freeflyer robots.

Code

Snake robot locomotion clip
Decentralized Control Strategy for Path-Following in Snake Robots
Arpit Dwivedi, Hemanta Hazarika, Dwaipayan Mukherjee, Debasattam Pal • CoDIT 2025 — Accepted

Designed a modular snake robot and proposed a decentralized path-following controller exploiting local link autonomy to track general curves smoothly.

Key Projects

Vision-Based End-to-End Planning for AVs
Transformer vision encoder + LSTM trajectories; intent-aware planning on Waymo Open Dataset.

Report

Vision-based pick-and-place manipulator
Vision-based Pick & Place Manipulator
Implemented a vision-based pick-and-place system that demonstrated object detection and manipulation capabilities. A U-Net-based segmentation model was implemented, achieving a mean Intersection over Union (mIoU) of 96% for accurate feature delineation in 2D images. Additionally, an end-to-end grasping algorithm utilizing grasping affordance maps was deployed to enhance object handling precision and reliability.
Gordon Botsy autonomous bot
Gordon Botsy: Autonomous Bot
Developed an autonomous bot capable of demonstrating reliability and efficiency in making a series of intelligent decisions to successfully complete the given challenge.

Code · Website

Genetic algorithm coverage graphic
Coverage Maximization for UAV Surveillance on Non-convex Domains using GA
Arpit Dwivedi, Chinmay Pimpalkhare • Advances in Multidisciplinary Design, Analysis and Optimization (NCMDAO), 2023

This work develops a UAV surveillance framework for disaster management, focusing on maximizing coverage in non-convex and disconnected regions. The framework processes input images, extracts boundaries, and reduces complexity by approximating space as a union of polygons. A genetic algorithm with Monte Carlo sampling optimizes sensor configurations, leveraging problem geometry for efficient coverage.

Code · Paper

Academic Service

Conference Reviewer: CDC 2025, CoDIT 2025, RSS (OOD Workshop) 2025