Senior/ Staff AI Engineer, Physical AI

Grid Dynamics; Kraków; On-site; PLN 240,000 - 420,000; Full time

Benefits offered by this job

Medical insurance

Sports

Professional development opportunities

Well-equipped office

Corporate social events

Job summary

Grid Dynamics, Nasdaq-listed tech services leader, seeks a Senior/Staff AI Engineer to advance multimodal generative models for robotics and physical AI. You will join the R&D team designing models for dexterous manipulation in manufacturing and logistics, lead research in perception, motion generation, and scalable AI pipelines.

The role demands strong Python, PyTorch and JAX/Flax skills, a track record of publications, and a PhD or MS in engineering or CS.

Qualifications

• Strong fundamentals and implementation skills in deep learning, vision or multimodal models.
• Proficient in Python and PyTorch; practical experience in ML model development.
• Experience with large-scale AI training (GPU clusters, Ray).
• Advanced degree or equivalent research experience in Engineering or CS.

Responsibilities

• Design and develop models for dexterous robotic manipulation for manufacturing and logistics.
• Lead research in robotic perception, motion generation, and physical AI.
• Develop algorithms and scalable AI pipelines for production use.
• Combine computer vision, motion planning, and RL/VLA models in robotic apps.
• Define data generation and supervise dataset scaling for model capability.

Skills

Deep learning

Vision models

Multimodal models

Python

PyTorch

JAX/Flax

Robotic manipulation

GPU clusters

Ray

Publications

Education

MS/PhD in Engineering or Computer Science

Tools

Ray

CUDA

JAX/Flax

Job description

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.