Senior AI Frameworks Communications Architect

NVIDIA; Austin (TX); On-site; USD 152,000 - 241,500; Full time

Job summary

A leading technology company located in Austin, Texas is seeking an experienced individual to integrate communication libraries with AI frameworks and enhance AI performance. The role requires 5+ years in software engineering, proficiency in deep learning frameworks like PyTorch, and adaptability in a dynamic environment. The salary ranges from $152,000 to $241,500 depending on experience, with additional equity and benefits offered. Interested candidates should apply by April 8, 2026.

Qualifications

• 5+ years of experience in software engineering and HPC/AI.
• Experience with deep learning frameworks such as PyTorch and JAX.
• Solid grasp of AI models and compiler technologies.

Responsibilities

• Integrate new communication libraries features in AI frameworks.
• Conduct in-depth AI workload performance characterization.
• Design fault-tolerant solutions for AI workloads.

Skills

Software engineering

AI frameworks

Deep Learning Frameworks

Python

C++

CUDA

Education

B.S., M.S. or Ph.D. in Computer Science or related field

Tools

PyTorch

NVIDIA Nsight Systems

Job description

A leading technology company located in Austin, Texas is seeking an experienced individual to integrate communication libraries with AI frameworks and enhance AI performance. The role requires 5+ years in software engineering, proficiency in deep learning frameworks like PyTorch, and adaptability in a dynamic environment. The salary ranges from $152,000 to $241,500 depending on experience, with additional equity and benefits offered. Interested candidates should apply by April 8, 2026.