GPU System Engineer

Artificial Intelligence Technologies LLC; Chicago (IL); Presencial; USD 150.000 - 210.000; Tempo integral

Resumo da vaga

Artifintech in Chicago is seeking expert-level GPU Software Engineers to support a high-visibility Maya program, building tooling on top of a custom compiler and SDK.

You will develop, optimize, and port GPU kernels and AI workloads to a specialized hardware platform, collaborating with platform architects and compiler teams to enhance end-to-end system performance.

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 High Performance LINPAC (HPL) and BERT 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

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 Description:

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.