EPAM Systems

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

Brak informacji o wynagrodzeniu

SeniorZdalnieFull-time

#390359·Dodano 4 miesiące temu·5

Źródło: EPAM Systems

Aplikuj teraz

Tech Stack / Keywords

C++Machine Learning PyTorchCUDA

Podsumowanie oferty

We are looking for a Lead ML&AI Engineer to join a team focused on building the next generation of computers for artificial intelligence. The team includes experts in computer architecture, ASIC design, and neural network compilers, working on the compiler and neural network stack, including LLMs and CNNs. This role involves developing kernels in C++ for AI and ML workloads, optimizing the compiler and neural network stack, and leading technical decisions within the ML & AI team.

Wymagania

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

Obowiązki

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

Benefity

Opportunity to work remotely within Poland and hybrid mode by design
Chance to work abroad for up to 60 days annually
Business-driven relocation opportunities
Career development programs
Thought leadership, mentoring, soft skills and well-being programs
Certification (Anthropic, Gemini, GCP, Azure, AWS)
English classes
Stable pay
Participation in the Employee Stock Purchase Plan with a 15% discount
Benefits package including health insurance, multisport, shopping vouchers
Referral bonuses up to $2,000
Offices with entertainment and relaxation zones, table tennis and football, free snacks, coffee
Corporate, social, and well-being events

Elastyczne godziny

Budżet konferencyjny

Dofinansowanie szkoleń

Kursy językowe

Karta sportowa

Opieka zdrowotna

Ubezpieczenie

Udziały pracownicze

Premie

Napoje w biurze

Darmowe przekąski

Spotkania integracyjne

EPAM Systems

1770 aktywnych ofert

Zobacz wszystkie oferty

Aplikuj teraz