Senior GPU Optimization Solutions Architect

Zymtrace; United States; Remote; USD 120,000 - 160,000; Full time

Benefits offered by this job

Competitive salary

401(k) plan

Comprehensive health, dental, and vision insurance

Annual learning and development budget

Job summary

A leading AI infrastructure optimization firm in the United States is seeking a Senior Solutions Architect. In this role, you will become the technical bridge between the company and key customers, focusing on GPU optimization for large-scale AI workloads. The ideal candidate has a strong background in solutions architecture, GPU computing, and hands-on experience with AI frameworks such as PyTorch or JAX. Join an innovative team to make a real impact on AI training and inference workloads.

Qualifications

• 2+ years in solutions architecture or technical customer-facing role.
• Strong understanding of CUDA, GPU memory, and performance bottlenecks.
• Experience with AI/ML frameworks like PyTorch and JAX.

Responsibilities

• Own the technical relationship with enterprise customers.
• Design and deliver technical demos and architecture reviews.
• Collaborate with engineering to improve the product.

Skills

Solutions Architecture

GPU Computing

AI/ML Frameworks

Technical Communication

Self-starter Mentality

Tools

CUDA

Linux Systems

Profiling Tools

Kubernetes

Job description

A leading AI infrastructure optimization firm in the United States is seeking a Senior Solutions Architect. In this role, you will become the technical bridge between the company and key customers, focusing on GPU optimization for large-scale AI workloads. The ideal candidate has a strong background in solutions architecture, GPU computing, and hands-on experience with AI frameworks such as PyTorch or JAX. Join an innovative team to make a real impact on AI training and inference workloads.