Senior GPU Software Engineer: High-Performance AI Kernels

Machine Learning Technologies, LLC; Chicago (IL); On-site; USD 140,000 - 190,000; Full time

Job summary

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.

Qualifications

• 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.

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 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.

Skills

GPU kernel programming

Torch

Triton

Open-source models

Benchmarking HPC

Tools

Custom compiler toolchains

Job description

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.