Solution Architect (Kernel Optimization & ML Performance)

Join us as a Solution Architect (Kernel Optimization & ML Performance) and play a key role in shaping impactful, scalable solutions that drive real business value . You will work at the intersection of advanced hardware acceleration and machine learning infrastructure, translating complex performance and optimization requirements into robust architectural designs. This role offers the opportunity to collaborate with global, cross-functional teams in a dynamic and fast-paced environment, working closely with ML researchers, compiler engineers, and systems architects.

You’ll have a direct influence on the technical direction of high-performance ML solutions, solution quality, and overall efficiency across large-scale AI workloads. At EPAM, we value innovation, ownership, and a proactive mindset, giving you the space to make a tangible technological impact. If you are ready to take on a strategic architect-level role in AI infrastructure and grow your career in a global setting, we encourage you to apply.

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 the client’s technical leadership, ML researchers, and engineers to capture requirements and design scalable solutions Evaluate and select the right technologies, toolchains, and design patterns to optimize compute-inte