GPU System Engineer

Machine Learning Technologies, LLC; Chicago (IL); Presencial; USD 140.000 - 190.000; Tempo integral

Resumo da vaga

Machine Learning Technologies, LLC in Chicago, IL seeks expert GPU Software Engineers to advance a high-visibility Maya program. You will develop and port GPU kernels on a specialized hardware platform, building tooling atop a custom compiler and SDK.

Responsibilities include optimizing kernels, porting AI workloads, and validating performance with benchmarks like HPL and BERT. You will collaborate with platform architects and system engineers to ensure end-to-end efficiency.

Qualificações

• Expert-level GPU kernel development for specialized hardware platforms.
• Experience with Torch and Triton frameworks.
• Porting AI workloads to proprietary SDK environments.
• Design kernel-level optimizations to improve hardware execution.
• Collaborate with platform architects and compiler teams.

Responsabilidades

• Develop high-performance GPU kernels for specialized hardware platforms using Torch and Triton frameworks.
• Build software solutions leveraging custom compiler toolchains and SDK capabilities.
• Design and implement kernel-level optimizations to control and improve hardware execution behavior.
• Port and optimize open-source AI/ML models for proprietary or custom SDK environments.
• Adapt and run high-performance computing benchmarks and stress workloads, including HPL and BERT.
• Develop stress testing and validation workloads aligned with hardware behavior and platform reliability goals.
• Support validation and performance testing for current and next-generation hardware platforms.
• Collaborate closely with platform architects, compiler teams, and system engineers to enhance end-to-end system performance.

Conhecimentos

GPU kernel programming

Torch

Triton

Open-source models

Benchmarking HPC

Ferramentas

Custom compiler toolchains

Descrição da vaga

US Citizen, Green Card, H-1B, OPT-EAD, CPT, Other

Work Authorization: USC, GC, H-1B, OPT-EAD, GC-EAD, H4-EAD

Job Summary

We are seeking expert-level GPU Software Engineers to support a high-visibility platform initiative within the Maya program, focused on building software tooling on top of a custom compiler and SDK.The role involves developing, optimizing, and porting GPU kernels and AI workloads to a specialized hardware platform.

Key Responsibilities

• Develop high-performance GPU kernels for specialized hardware platforms using Torch and Triton frameworks
• Build software solutions leveraging custom compiler toolchains and SDK capabilities
• Design and implement kernel-level optimizations to control and improve hardware execution behavior
• Port and optimize open-source AI/ML models for proprietary or custom SDK environments
• Adapt and run high-performance computing benchmarks and stress workloads, including:
• High Performance LINPAC (HPL)
• BERT and similar benchmark-style workloads
• Develop stress testing and validation workloads aligned with hardware behavior and platform reliability goals
• Support validation and performance testing for both current and next-generation hardware platforms
• Collaborate closely with platform architects, compiler teams, and system engineers to enhance end-to-end system performance

We are an equal opportunity employer. All aspects of employment including the decision to hire, promote, discipline, or discharge, will be based on merit, competence, performance, and business needs. We do not discriminate on the basis of race, color, religion, marital status, age, national origin, ancestry, physical or mental disability, medical condition, pregnancy, genetic information, gender, sexual orientation, gender identity or expression, national origin, citizenship/ immigration status, veteran status, or any other status protected under federal, state, or local law.