Senior/ Staff AI Engineer, Physical AI

We are looking for a highly skilled and passionate Senior/Staff AI Engineer with expertise in multimodal generative models to join our team. As a Senior/Staff AI Engineer, you will play a key role in developing and advancing cutting-edge solutions in the field of robotics and physical AI. If you have exceptional AI engineering skills and want to shape the future of AI in the physical world, we’d love to connect.

Responsibilities

• You will be a part of the RnD team, designing and developing models for dexterous robotic manipulation for Manufacturing and Logistics applications
• Lead and contribute to cutting-edge research in robotic perception, motion generation, and physical AI
• Design, develop, and experiment with novel algorithms, architectures, and techniques that enhance the performance, efficiency, and scalability of our AI models
• Design and implement robotic applications combining conventional computer vision, motion planning, and VLA/RL models
• Define data generation, collection, and curation strategies; supervise data collection to scale up our datasets and model capabilities

Requirements

• Excellent fundamentals and excellent implementation skills in deep learning, vision models, or multimodal models
• Strong engineering ability with Python and PyTorch, experience building ML models from scratch in PyTorch and/or JAX/Flax
• Excellent system design skills and experience with complex collaborative AI projects, ability to implement research ideas in a clean way and productize AI pipelines
• Experience with robotic manipulation (manufacturing or logistics applications) or other physical AI projects (ADAS, autonomous driving, etc.) is a big plus
• Experience working with large scale AI training (GPU clusters, Ray)
• Strong publication record in deep learning, generative models, computer vision, or physical AI (RL, VLA)
• MS or PhD degree in Engineering or Computer Science or equivalent experience

We offer

• Opportunity to work on bleeding-edge projects
• Work with a highly motivated and dedicated team
• Competitive salary
• Flexible schedule
• Benefits package - medical insurance, sports
• Corporate social events
• Professional development opportunities
• Well-equipped office

About Us

Grid Dynamics (NASDAQ: GDYN) is a leading provider of technology consulting, platform and product engineering, AI, and advanced analytics services. Fusing technical vision with business acumen, we solve the most pressing technical challenges and enable positive business outcomes for enterprise companies undergoing business transformation. A key differentiator for Grid Dynamics is our 8 years of experience and leadership in enterprise AI, supported by profound expertise and ongoing investment in data, analytics, cloud & DevOps, application modernization and customer experience. Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the Americas, Europe, and India.