Senior Solution Architect - ML Kernel & Performance

EPAM Systems; Town of Poland (NY); On-site; USD 140,000 - 180,000; Full time

Job summary

EPAM Systems is seeking a Solution Architect specializing in Kernel Optimization and ML Performance. This role requires extensive experience in software engineering and architecture, focusing on high-performance machine learning solutions. As a key member, you’ll define architectures for ML workloads, collaborate with global teams, and mentor development teams.

The ideal candidate will have a Bachelor’s or Master’s degree in Computer Science, along with in-depth knowledge of machine learning frameworks and performance optimization techniques.

Qualifications

• 12+ years of software engineering experience, including 5+ years in architecture or technical leadership roles.
• In-depth knowledge of ML frameworks and Core ML concepts.
• Hands-on expertise in performance optimization at the kernel level targeting TPUs/GPUs.

Responsibilities

• Define and own the architecture for performance-critical ML workloads.
• Create a strategic roadmap for kernel optimization and performance improvement.
• Provide technical leadership and mentorship to development teams.

Skills

ML frameworks (JAX, PyTorch, TensorFlow)

Performance optimization

C++/Python for high-performance computing

Communication skills

Education

Bachelor’s or Master’s degree in Computer Science

Tools

MLIR

OpenXLA

Job description

EPAM Systems is seeking a Solution Architect specializing in Kernel Optimization and ML Performance. This role requires extensive experience in software engineering and architecture, focusing on high-performance machine learning solutions. As a key member, you’ll define architectures for ML workloads, collaborate with global teams, and mentor development teams.

The ideal candidate will have a Bachelor’s or Master’s degree in Computer Science, along with in-depth knowledge of machine learning frameworks and performance optimization techniques.