Senior Solution Architect: Kernel & ML Performance

EPAM Systems; Kraków; Hybrid; PLN 400,000 - 640,000; Full time

Benefits offered by this job

Hybrid by design

Remote work within Poland

Relocation opportunities

Stock purchase program or equivalent

Job summary

EPAM Systems is seeking a Solution Architect to lead kernel optimization and ML performance across TPU/GPU workloads. You will define architectures, roadmap optimization, and collaborate with ML researchers, compiler engineers, and system architects.

The role offers exposure to global teams, hybrid work within Poland, and opportunities to influence high‑performance ML infrastructure with emphasis on security, scalability, and production readiness.

Qualifications

• Bachelor’s or Master’s degree in Computer Science or equivalent practical experience.
• 12+ years of software engineering experience, including 5+ years in architecture or technical leadership roles.
• Hands-on experience with performance optimization at kernel level for ML workloads.
• Strong experience with C++/Python for high-performance computing.

Responsibilities

• Define and own the architecture for performance-critical ML workloads, leveraging custom kernels on TPUs and GPUs.
• Create a strategic roadmap for kernel optimization, framework integration, and large-scale performance improvement.
• Collaborate with client’s technical leadership, ML researchers, and engineers to capture requirements and design scalable solutions.
• Evaluate and select technologies, toolchains, and patterns to optimize compute-intensive operations.
• Guide benchmarking infrastructure, autotuning, profiling tools, and regression suites.
• Advocate ML performance best practices across teams.
• Provide technical leadership and mentorship to development teams implementing architectural solutions.
• Ensure security, scalability, and maintainability in design.

Skills

JAX

PyTorch

TensorFlow

Core ML

C++

Python

MLIR

OpenXLA

Performance optimization

Kernel programming

Education

Bachelor's or Master's in Computer Science

Tools

MLIR

OpenXLA

Job description

EPAM Systems is seeking a Solution Architect to lead kernel optimization and ML performance across TPU/GPU workloads. You will define architectures, roadmap optimization, and collaborate with ML researchers, compiler engineers, and system architects.

The role offers exposure to global teams, hybrid work within Poland, and opportunities to influence high‑performance ML infrastructure with emphasis on security, scalability, and production readiness.