Lead C++ Developer w Kraków, Polska

Expected, C++, Python

About the project, We are seeking a talented Lead C++ Developer to join the highly interdisciplinary CoreML team, where you will drive the performance and optimization of both training and serving, delivering massive impact for customers. In this role, you will have exposure to the newest Tensor Processing Unit (TPU), Graphics Processing Unit (GPU) hardware, the latest ML models, and the advanced toolchains that bridge them. Your work will directly enable AI research, production deployments and the broader open-source ecosystem, addressing complex technical issues that directly impact the efficiency and scalability of AI across the industry.

Your responsibilities, Design and optimize high-performance kernels (using languages like Pallas, Mosaic and Triton) targeting Tensor Processing Unit (TPU) and Graphics Processing Unit (GPU) architectures for critical Machine Learning (ML) operations, redefining what's possible from massive training runs to high-speed inference, Architecture of infrastructure such as benchmarking suites, autotuning frameworks, performance analysis tools, regression testing and documentation, transforming how the developer community interacts with increasingly critical custom kernels in key Open-Source Software (OSS) libraries, Track the latest advancements in hardware architectures, compiler technologies and AI models to identify new opportunities for performance optimization through custom kernels, Engagement with ML researchers, framework developers (Just After eXecution (JAX), PyTorch) and compiler engineers (Accelerated Linear Algebra (XLA)) to enhance adoption, identify new requirements and address bottlenecks by providing appropriate solutions

Bachelor's degree or equivalent practical experience, Overall 7+ years of industry experience, 5 years of experience with software development in C++ or Python, 3 years of experience testing, maintaining or launching software products, and 1 year of experience with software design and architecture, LLM experience, Expertise in performance optimization at the kernel level, English proficiency at B2 level or higher

Optional, Skills in optimizing TPU/GPU code, using low-level kernel languages like Pallas, Compute Unified Device Architecture (CUDA) or Triton, Knowledge of ML Frameworks (JAX/PyTorch), common operations like attention and Mixture of Experts (MoEs), including model optimization and low-precision formats, Understanding of modern accelerators (e.g., data movement, pipelining, heterogeneous compute and scale-out), Understanding of compiler principles (optimization, code generation) and toolchains such as MLIR and OpenXLA, Showcase of building developer infrastructure, including Open-Source Software (OSS) libraries, flexible high-performance APIs and easy-to-consume documentation to empower the community

What we offer, We gather like-minded people:, •Top tech minds driving innovation in AI, cloud and digital platform modernization, •Supportive team and agile, startup-like culture, •Hybrid by design mode and opportunity to work remotely within Poland, •Chance to work abroad for up to 60 days annually, •Business-driven relocation opportunities, , We provide growth opportunities:, •Career development programs, •Thought leadership, mentoring, soft skills and well-being programs, •Certification (Anthropic, Gemini, GCP, Azure, AWS), •English classes, , We cover it all:, •Stable pay, •Participation in the Employee Stock Purchase Plan with a 15% discount, •Benefits package (health insurance, multisport, shopping vouchers), •Referral bonuses up to $2,000, •Offices featuring entertainment and relaxation zones, table tennis and football, free snacks, coffee and more, •Corporate, social and well-being events, , Please, note:, •Benefits listed above are available to employees only, •We are open for working with Contractors. Terms of B2B cooperation agreements are agreed individually, •We will reach out to selected candidates exclusively

This is how we work on a project, Continuous Deployment, Continuous Integration

Development opportunities we offer, conferences in Poland, development budget, external training, industry-specific e-learning platforms, intracompany training, mentoring, soft skills training, space for experimenting, substantive support from technological leaders, support of IT events, technical knowledge exchange within the company

Benefits, private medical care, life insurance, flexible working time, fruits, integration events, retirement pension plan, no dress code, video games at work, coffee / tea, parking space for employees, leisure zone, employee referral program, charity initiatives

Recruitment stages, Recruiter Interview (30 min), Technical Interview (90 min), Manager Interview (60 min), Meeting with the Project Team or Client (60 min, optional), Welcome Aboard!

EPAM Systems (Poland) sp. z o.o., EPAM is a leading global provider of digital platform engineering and development services. We are committed to having a positive impact on our customers, our employees, and our communities. We embrace a dynamic and inclusive culture. Here you will collaborate with multi-national teams, contribute to a myriad of innovative projects that deliver the most creative and cutting-edge solutions, and have an opportunity to continuously learn and grow. No matter where you are located, you will join a dedicated, creative, and diverse community that will help you discover your fullest potential.