"Alerted.org

Job Description

As a Senior AI Engineer, you’ll be a core member of a pioneering team focused on developing foundational AI infrastructure tailored for power grid applications. You’ll play a key role in architecting and refining the systems, data pipelines, and training workflows that support advanced AI research. Collaborating closely with research scientists, you’ll help transform innovative ideas into scalable, high-performance implementations that accelerate the deployment of impactful AI solutions. This position demands hands-on expertise in distributed training, data engineering, and MLOps, along with a strong track record of building resilient and scalable AI systems. The ideal candidate is proactive, resourceful, and committed to delivering high-quality solutions with speed and precision.

Job Responsibilities

• Architect and optimize large-scale training and fine-tuning workflows, from data ingestion to inference, with a focus on maximizing Model Flop Utilization (MFU) across compute clusters.

• Collaborate closely with research teams to convert experimental models and algorithms into efficient, production-ready code.

• Identify and resolve performance bottlenecks throughout the training stack, continuously improving speed and scalability.

• Evaluate and select hardware, software, and cloud technologies to support the AI infrastructure platform.

• Implement MLOps best practices using tools like MLFlow and Weights & Biases to ensure reproducibility, reliability, and continuous model improvement.

• Maintain comprehensive documentation of infrastructure and training processes, and stay informed on emerging strategies to enhance workflows and system performance.

We are a company committed to creating diverse and inclusive environments where people can bring their full, authentic selves to work every day. We are an equal opportunity/affirmative action employer that believes everyone matters. Qualified candidates will receive consideration for employment regardless of their race, color, ethnicity, religion, sex (including pregnancy), sexual orientation, gender identity and expression, marital status, national origin, ancestry, genetic factors, age, disability, protected veteran status, military or uniformed service member status, or any other status or characteristic protected by applicable laws, regulations, and ordinances. If you need assistance and/or a reasonable accommodation due to a disability during the application or recruiting process, please send a request to [email protected] learn more about how we collect, keep, and process your private information, please review Insight Global's Workforce Privacy Policy: https://insightglobal.com/workforce-privacy-policy/.

Skills and Requirements

• Master’s degree or higher in Computer Science, Engineering, or a related technical discipline.

• Minimum of 5 years of experience as a Data & AI Engineer or Machine Learning Engineer, with a focus on infrastructure for large-scale machine learning systems. Candidates with more or less experience may be considered for different levels.

• Strong hands-on experience with AI frameworks such as PyTorch, JAX, or PyTorch Lightning, and expertise in multi-node GPU training and optimization for large foundation models.

• Proven ability to troubleshoot, debug, and optimize performance using data-driven approaches.

• Solid communication and collaboration skills, with experience implementing MLOps practices for model tracking, evaluation, and deployment.' • Active GitHub profile showcasing open-source contributions to data engineering or deep learning infrastructure projects.

• Experience developing custom CUDA, Triton, or CUTLASS kernels.

• Familiarity with performance monitoring and profiling tools for distributed training and data pipeline null

We are a company committed to creating diverse and inclusive environments where people can bring their full, authentic selves to work every day. We are an equal employment opportunity/affirmative action employer that believes everyone matters. Qualified candidates will receive consideration for employment without regard to race, color, ethnicity, religion,sex (including pregnancy), sexual orientation, gender identity and expression, marital status, national origin, ancestry, genetic factors, age, disability, protected veteran status, military oruniformed service member status, or any other status or characteristic protected by applicable laws, regulations, andordinances. If you need assistance and/or a reasonable accommodation due to a disability during the application or the recruiting process, please send a request to [email protected].