Home
Communities
Companies
- Companies
  
  Discover best places to work
- Compare Companies
  
  Compare & find best workplace
- Add Office Photos
  
  Bring your workplace to life
- Add Company Benefits
  
  Highlight your company's perks
Reviews
- Company reviews
  
  Read reviews for 6L+ companies
- Write a review
  
  Rate your former or current company
Salaries
- Browse salaries
  
  Discover salaries for 6L+ companies
- Salary calculator
  
  Calculate your take home salary
- Are you paid fairly?
  
  Check your market value
- Share your salary
  
  Help other jobseekers
- Gratuity calculator
  
  Check your gratuity amount
- HRA calculator
  
  Check how much of your HRA is tax-free
- Salary hike calculator
  
  Check your salary hike
Interviews
- Company interviews
  
  Read interviews for 40K+ companies
- Campus placements
  
  Interviews questions for 2K+ colleges
- Share interview questions
  
  Contribute your interview questions
Jobs
Awards

WINNERS AWAITED!
- ABECA 2025
  
  WINNERS AWAITED!
  
  AmbitionBox Employee Choice Awards - 4th Edition
- ABECA 2024
  
  AmbitionBox Employee Choice Awards - 3rd Edition
- AmbitionBox Best Places to Work 2022
  
  2nd Edition
- AmbitionBox Best Places to Work 2021
  
  1st Edition

Premium Employer

Siemens

Compare

4.1

based on 4.8k Reviews

691 Siemens Jobs

Senior AI Engineer

Siemens Limited

4.1

based on 4.8k Reviews

5-10 years

Gurgaon / Gurugram

1 vacancy

Senior AI Engineer

Siemens

posted 9d ago

Job Role Insights

Flexible timing

Key skills for the job

Research Machine Learning Github Training Agile Coaching Open Source

+ 2 more

Job Description

We are seeking a highly skilled and driven Senior AI Engineer to join our team as a founding member, developing the critical data and AI infrastructure for training foundation models for power grid applications. You will be instrumental in building and optimizing the end-to-end systems, data pipelines, and training processes that will power our AI research. Working closely with research scientists, you will translate cutting-edge research into robust, scalable, and efficient implementations, enabling the rapid development and deployment of transformational AI solutions. This role requires deep hands-on expertise in distributed training, data engineering, MLOps, a proven track record of building scalable AI infrastructure.

Your new role - challenging and future- oriented:

Design, build, and rigorously optimize everything necessary for large-scale training, fine-tuning and/or inference with different model architectures. Includes the complete stack from dataloading to distributed training to inference; to maximize the MFU (Model Flop Utilization) on the compute cluster.
Collaborate closely and proactively with research scientists, translating research models and algorithms into high-performance, production-ready code and infrastructure. Ability to implement, integrate & test latest advancements from research publications or open-source code.
Relentlessly profile and resolve training performance bottlenecks, optimizing every layer of the training stack from data loading to model inference for speed and efficiency.
Contribute to technology evaluations and selection of hardware, software, and cloud services that will define our AI infrastructure platform.
Experience with MLOps frameworks (MLFlow, WnB, etc) to implement best practices across the model lifecycle - development, training, validation, and monitoring - ensuring reproducibility, reliability, and continuous improvement.
Create thorough documentation for infrastructure, data pipelines, and training procedures, ensuring maintainability and knowledge transfer within the growing AI lab.
Stay at the forefront of advancements in large-scale training strategies and data engineering and proactively driving improvements and innovation in our workflows and infrastructure.
High-agency individual demonstrating initiative, problem-solving, and a commitment to delivering robust and scalable solutions for rapid prototyping and turnaround.

We don't need superheroes, just super minds:

Bachelors or masters degree in computer science, Engineering, or a related technical field.
5+ years of hands-on experience in a role specifically building and optimizing infrastructure for large-scale machine learning systems
Deep practical expertise with AI frameworks (PyTorch, Jax, Pytorch Lightning, etc). Hands-on experience with large-scale multi-node GPU training, and other optimization strategies for developing large foundation models, across various model architectures. Ability to scale solutions involving large datasets and complex models on distributed compute infrastructure.
Excellent problem-solving, debugging, and performance optimization skills, with a data-driven approach to identifying and resolving technical challenges.
Strong communication and teamwork skills, with a collaborative approach to working with research scientists and other engineers.
Experience with MLOps best practices for model tracking, evaluation and deployment.

Desired skills

Public GitHub profile demonstrating a track record of open-source contributions to relevant projects in data engineering or deep learning infrastructure is a BIG PLUS.
Experience with performance monitoring and profiling tools for distributed training and data pipelines.
Experience writing CUDA/Triton/CUTLASS kernels.

Employment Type: Full Time, Permanent

Read full job description

Prepare for Senior Artificial Intelligence Engineer roles with real interview advice

Top Siemens Senior Artificial Intelligence Engineer Interview Questions

Q1. Count Inversions Problem Statement Given an integer array ARR of size N , your task is to find the total number of inversions that exist in ... read more

asked in Software Developer interview

View answer (1)

Q2. Maximum Subarray Sum Problem Statement Given an array arr of length N consisting of integers, find the sum of the subarray (including empty ... read more

asked in Software Developer interview

View answer (1)

Q3. Remove Character from String Problem Statement Given a string str and a character 'X', develop a function to eliminate all instances of 'X' ... read more

asked in Software Developer interview

View answer (1)

View all 370 questions