GPU Software Architect for AI Stack & ROCm

Triune Infomatics Inc

California

On-site

USD 150,000 - 210,000

Full time

18 hours ago

Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Triune Infomatics Inc is seeking an experienced GPU Software Engineer/Architect for a 12-month milestone-based engagement to support a cutting-edge GPU software integration project on AMD MI210 platforms. You will drive AI stack development, contribute to open-source projects, and deliver performance benchmarks and integration reports across monthly milestones.

Applicants should have deep expertise in ROCm, GPU architecture, SGLang, and end-to-end systems integration, with a strong track record

Qualifications

Hands-on experience with AMD MI210 GPUs and ROCm stack.
Experience with GPU kernel programming and memory hierarchy optimization.
Familiarity with SGLang and LLM inference frameworks.
Open-source contribution workflows (GitHub, PRs, reviews) and project management tools.

Responsibilities

Design and develop GPU software modules aligned with monthly milestones.
Perform end-to-end testing of AI stack software modules.
Validate AMD Infinity Bridge and AIS on MI210 GPU hardware.
Conduct functional and performance benchmarking (pSLC, ROCm).
Develop and contribute CaMa and SGLang changes to ROCm stack and open-source community.
Publish integration docs and benchmarking reports.