Senior C++ Software Engineer with CUDA/GPU/TPU

EPAM Systems; Poland; Hybrid; PLN 240,000 - 360,000; Full time

Benefits offered by this job

Hybrid by design

Remote work within Poland

Occasional international travel

Relocation opportunities

Employee Stock Purchase Plan with 15%

Referral bonuses up to $2,000

Job summary

EPAM Systems seeks a Senior C++ Software Engineer to join our ML & AI team building next-gen AI computing platforms. You will write high-performance kernels, optimize the compiler and neural network stack for LLMs and CNNs, and collaborate with experts in architecture and ASIC design.

You should have 5+ years in C++/CUDA, strong ML model experience with PyTorch or TensorFlow, and English at B2 or higher. Hybrid work in Poland with potential international travel and relocation opportunities are

Qualifications

• 5+ years of software engineering experience in ML/AI.
• 5+ years of experience in C++ and CUDA/kernel programming.
• Experience with GPU/AI-accelerator architectures.
• Experience with PyTorch or TensorFlow.
• English proficiency at B2 level or higher.

Responsibilities

• Write kernels in C++ for AI/ML workloads.
• Develop and optimize the compiler and neural network stack for LLMs/CNNs.
• Collaborate with experts in architecture/ASIC design.
• Design AI/ML kernel ops for GPU/accelerator architectures.
• Build and maintain ML models using PyTorch or TensorFlow.
• Ensure kernel performance and correctness across NN architectures.
• Take ownership of key technical decisions as lead developer.

Skills

C++

CUDA/kernel programming

ML/AI

PyTorch

TensorFlow

English (B2)

Tools

PyTorch

TensorFlow

Job description

We are looking for a Senior C++ Software Engineer to join a team building the next generation of computers for artificial intelligence. The company brings together experts in computer architecture, ASIC design, advanced systems and neural network compilers, with the team focused on the compiler and neural network stack, including LLMs and CNNs. This is a key developer role within the ML & AI team.

Responsibilities

• Write kernels in C++ for various operations used across AI and ML workloads
• Develop and optimize the compiler and neural network stack supporting LLMs and CNNs
• Collaborate with experts in computer architecture, ASIC design and advanced systems to build next-generation computing solutions
• Design and implement AI/ML kernel operations tailored to GPU/AI-accelerator architectures
• Build and maintain ML models using PyTorch or TensorFlow
• Ensure kernel-level performance and correctness across modern neural network architectures such as transformers
• Take ownership of key technical decisions as a lead developer within the ML & AI team

Requirements

• 5+ years of experience in software engineering with a focus on ML/AI
• 5+ years of experience in C++ and CUDA/kernel programming
• Knowledge in GPU/AI-accelerator architectures
• Experience in building and working with ML models in PyTorch or TensorFlow
• Understanding of modern ML model architectures such as transformers
• English proficiency at B2 level or higher

Nice to have

• Proficiency in Python programming with hands-on experience in developing, training and fine-tuning deep learning models
• Skills in designing, implementing and iterating on neural network architectures to achieve optimal performance on diverse tasks
• Capability to investigate and troubleshoot model performance issues including gaps in compilation steps and kernels
• Demonstrated experience in designing, training and deploying neural networks for various applications
• Solid understanding of machine learning fundamentals including supervised and unsupervised learning techniques

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
• Thought leadership, mentoring, soft skills and well-being programs
• We cover it all:
• Participation in the Employee Stock Purchase Plan with a 15% discount
• 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

EPAM is global leader in AI transformation engineering and integrated consulting, serving Forbes Global 2000 companies and ambitious startups. With over thirty years of expertise in custom software, product and platform engineering, we empower our clients to become AI-Native enterprises, driving measurable value from innovation and digital investments.