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

À propos de la fonction

Emplacement
UNITED STATES OF AMERICA
Florida
Orlando
Télétravail ou au bureau
Hybride (télétravail/bureau)
Entreprise
Siemens Energy, Inc.
Organisation
Grid Technologies
Unité opérationnelle
Digital Grid
Temps plein / partiel
Temps plein
Niveau d’expérience
Expérimenté


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

#LI-CDS


Déclaration sur l'égalité des chances en matière d'emploi


Siemens Energy est un employeur qui souscrit au principe de l'égalité des chances et de l'action positive, et qui encourage la diversité sur le lieu de travail. Tous les candidats qualifiés seront pris en considération pour un emploi sans égard aux éléments suivants : race, couleur de peau, croyance, religion, pays d'origine, statut de citoyen, ascendance, sexe, âge, handicap physique ou mental non lié aux capacités, état matrimonial, responsabilités familiales, grossesse, information génétique, orientation sexuelle, expression de genre, identité de genre, personne transgenre, stéréotypes sexuels, ordonnance de protection, statut d'ancien combattant protégé ou militaire, ou révocation défavorable du service militaire, et autres catégories protégées par la loi fédérale, nationale ou
locale.

L'égalité des chances en matière d'emploi fait partie de la loi
Les candidats et les employés sont protégés par la loi fédérale contre la discrimination. Cliquez ici pour en savoir plus.

Disposition de non-discrimination sur la transparence salariale
Siemens se conforme au décret américain n° 11246, qui comprend la disposition de non-discrimination sur la transparence salariale. Cliquez ici pour en savoir plus (.PDF).

Loi californienne sur la protection des données
Les résidents de Californie ont le droit de recevoir des avis supplémentaires concernant leurs informations personnelles. Cliquez ici pour en savoir plus.