Ahmad Rahimi

PhD student at VITA, EPFL

prof_pic.jpg

I am a PhD student at the VITA lab at EPFL advised by Prof. Alexandre Alahi. My research interests include deep learning and computer vision for autonomous vehicles. I have experience in vehicle trajectory prediction methods, video generation and world models.

I got my B.Sc. in Computer Science from Sharif University of Technology, Iran, in 2022.

news

May 09, 2026 My internship at Wayve in London is coming to an end. It was a great experience and I learned a lot about the industry and the challenges of autonomous driving. I am grateful for the opportunity and the support of my colleagues. They are truly an exceptional team and I am proud to have been part of it. Thank you, Wayve!
Feb 21, 2026 My recent project on motion appearance decoupling for driving world models was accepted at CVPR 2026!
Nov 24, 2025 I joined Wayve as a Research Scientist, working on GAIA-related projects. Very excited to be part of this amazing team and contribute to the future of autonomous driving!
Feb 26, 2025 Two of my papers got accepted at CVPR 2025! Cheers 🍺!
Sep 15, 2022 I joined VITA Lab to do a PhD in Computer Science at EPFL. Very exciting!

selected publications

  1. CVPR
    mad-pull_video.gif
    MAD: Motion Appearance Decoupling for efficient Driving World Models
    Ahmad Rahimi*, Valentin Gerard*, Eloi Zablocki, and 2 more authors
    2026
    arXiv Code Website
  2. CVPR
    gem-control.gif
    GEM: A Generalizable Ego-vision Multimodal World Model for Fine-Grained Ego-Motion, Object Dynamics, and Scene Composition Control
    Mariam Hassan*, Sebastian Stapf*, Ahmad Rahimi*, and 17 more authors
    2025
    arXiv Code Website
  3. arxiv
    offroad-loss.png
    A Multi-Loss Strategy for Vehicle Trajectory Prediction: Combining Off-Road, Diversity, and Directional Consistency Losses
    Ahmad Rahimi, and Alexandre Alahi
    2024
    arXiv Code
  4. CVPR
    social-causality.jpg
    Sim-to-Real Causal Transfer: A Metric Learning Approach to Causally-Aware Interaction Representations
    Ahmad Rahimi, Po-Chien Luan, Yuejiang Liu, and 2 more authors
    2025
    arXiv Code
  5. CVPR
    scene-attack.gif
    Vehicle trajectory prediction works, but not everywhere
    Mohammadhossein Bahari, Saeed Saadatnejad, Ahmad Rahimi, and 4 more authors
    In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2022
    arXiv Code Website