Senior AI Engineer - Foundation Model Training & Infrastructure for Power Grids (US)

A Snapshot of Your Day

As a highly skilled and driven Senior AI Engineer, you’d be a founding team 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. Our successful candidate will be a high-agency individual demonstrating initiative, problem-solving, and a commitment to delivering robust and scalable solutions for rapid prototyping and turnaround.

How You’ll Make An Impact

- Designing, building, and optimizing all aspects of large-scale training and fine-tuning, from dataloading to inference, to maximize Model Flop Utilization (MFU) on large compute clusters.

- Working closely and proactively with research scientists to translate models and algorithms into high-performance, production-ready code, integrating and testing the latest advancements.

- Relentlessly profiling and resolving training performance bottlenecks, optimizing the entire training stack for speed and efficiency.

- Contributing to the technology evaluations and selection of hardware, software, and cloud services for the AI infrastructure platform.

- Using MLOps frameworks (MLFlow, WnB, etc.) to ensure best practices across the model lifecycle, ensuring reproducibility, reliability, and continuous improvement.

- Creating thorough documentation for infrastructure and training procedures, staying updated on advancements in training strategies, and driving improvements in workflows and infrastructure.

What You Bring

- Master's degree or higher in Computer Science, Engineering, or a related technical field.

- 5 or more years in a Data & AI (Artificial Intelligence) Engineer or Machine Learning Engineer, focusing on building and optimizing infrastructure for large-scale machine learning systems. *Candidates with more experience can be considered for a higher level or vice-versa.

- Deep practical expertise with AI frameworks (PyTorch, Jax, Pytorch Lightning, etc.), large-scale multi-node GPU training, and optimization strategies for large foundation 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, experience with MLOps best practices for model tracking, evaluation, and deployment.

- Preferred qualifications include: Public GitHub profile with a track record of open-source contributions to data engineering or deep learning infrastructure projects, experience writing CUDA/Triton/CUTLASS kernels, and proficiency with performance monitoring and profiling tools for distributed training and data pipelines.

About The Team

Join Siemens Energy new AI initiative to transform power grids. At our newly established AI Lab with powerful dedicated compute cluster, we are tackling the critical challenge of global grid congestion and revolutionizing antiquated grid operation and planning. We're combining 100 years of energy domain expertise with the latest advancements in AI foundation models for energy applications to fundamentally reshape the future of substations and power grids.

The AI Lab will be pioneering innovative and transformational solutions that will be productized and deployed across a large global footprint. This is your chance to drive a paradigm shift in the energy sector and build a more resilient and sustainable energy infrastructure.

Who is Siemens Energy?

At Siemens Energy, we are more than just an energy technology company. With ~100,000 dedicated employees in more than 90 countries, we develop the energy systems of the future, ensuring that the growing energy demand of the global community is met reliably and sustainably. The technologies created in our research departments and factories drive the energy transition and provide the base for one sixth of the world's electricity generation.

Our global team is committed to making sustainable, reliable, and affordable energy a reality by pushing the boundaries of what is possible. We uphold a 150-year legacy of innovation that encourages our search for people who will support our focus on decarbonization, new technologies, and energy transformation.

Find out how you can make a difference at Siemens Energy: https://www.siemens-energy.com/employeevideo

Rewards/Benefits

- Career growth and development opportunities

- Supportive work culture

- Company paid Health and wellness benefits

- Paid Time Off and paid holidays

- 401K savings plan with company match

- Family building benefits and parental leave

https://jobs.siemens-energy.com/jobs