Senior Multi-GPU AI Frameworks Engineer

NVIDIA; Westford (MA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

A leading tech company in Westford, Massachusetts is seeking an experienced software engineer specializing in AI frameworks. Candidates should have a strong background in deep learning and HPC, with proven experience in Python and C++. The role involves integrating new features in AI libraries, conducting performance analysis, and designing solutions for AI workloads. The expected salary ranges from $152,000 to $287,500 depending on experience, and the position offers equity and various benefits.

Qualifications

• 5+ years software engineering experience and HPC/AI experience.
• Development or integration experience with frameworks like PyTorch.
• Solid grasp of parallelism and compiler technologies.

Responsibilities

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

Skills

Python

C++

CUDA

Deep Learning Frameworks

AI performance benchmarking

Communication concepts

Education

B.S, M.S., or PhD in Computer Science

Tools

Performance profiler tools

Job description

A leading tech company in Westford, Massachusetts is seeking an experienced software engineer specializing in AI frameworks. Candidates should have a strong background in deep learning and HPC, with proven experience in Python and C++. The role involves integrating new features in AI libraries, conducting performance analysis, and designing solutions for AI workloads. The expected salary ranges from $152,000 to $287,500 depending on experience, and the position offers equity and various benefits.