Lead AI Engineer with Machine Learning

EPAM Systems, Inc.; Turkey; On-site; TRY 5,395,000 - 8,828,000; Full time

Benefits offered by this job

Continuous upskilling

Learning & development

Diversity of tasks

Mentoring programs

Learning platforms

English courses

Extra leave days

Referral bonuses

Private health insurance

Well-being events

IT Equipment

HUP program

Job summary

EPAM Systems, Inc. seeks a Lead AI Engineer to join a cutting‑edge team building AI-focused hardware and software platforms. You will design, train, and optimize neural networks using PyTorch, and map models onto specialized hardware through compilation steps and kernels.

You will lead engineers, collaborate with teams on compilers, NN, LLMs, and CNNs, and deploy scalable AI solutions across diverse applications while communicating clearly in English.

Qualifications

• 5+ years of experience in machine learning and AI engineering.
• At least 1 year of relevant leadership experience.
• Proficiency in Python with hands-on PyTorch for developing and training deep learning models.
• Ability to map models onto specialized AI hardware through compilation steps and kernels.

Responsibilities

• Design, implement, and iterate on neural network architectures using PyTorch to achieve optimal performance on diverse tasks.
• Train, validate, and fine-tune machine learning models using relevant datasets to ensure high accuracy and robustness.
• Investigate and troubleshoot model performance issues, iterating on models and techniques to continuously improve results.
• Analyze how models map onto Tenstorrent devices through compilation steps and kernels.
• Collaborate with cross-functional teams working on compilers, neural networks, LLMs, and CNNs.
• Deploy neural networks for various applications, ensuring scalability and reliability.

Skills

Python

PyTorch

Leadership

English communication

C++

Kernel programming

GPU architectures

Job description

We are seeking a Lead AI Engineer to join a cutting-edge team building computers designed specifically for AI. Our customer brings together experts in computer architecture, ASIC design, advanced systems, and neural network compilers to build the next generation of computing. The team is actively working on compilers, neural networks, LLMs, and CNNs, and is looking for an experienced engineer to take on a Key Developer role within the ML & AI team.ResponsibilitiesDesign, implement, and iterate on neural network architectures using PyTorch to achieve optimal performance on diverse tasksTrain, validate, and fine-tune machine learning models using relevant datasets to ensure high accuracy and robustnessInvestigate and troubleshoot model performance issues, iterating on models and techniques to continuously improve resultsAnalyze how models map onto Tenstorrent devices through compilation steps and kernelsIdentify gaps in functionality and performance, proposing innovative solutions to mitigate any issuesCollaborate with cross-functional teams working on compilers, neural networks, LLMs, and CNNsDeploy neural networks for various applications, ensuring scalability and reliabilityRequirements5+ years of experience in machine learning and AI engineeringAt least 1 year of relevant leadership experienceProficiency in Python programming with hands-on experience in PyTorch for developing and training deep learning modelsExpertise in designing, training, and deploying neural networks for various applicationsUnderstanding of machine learning fundamentals, including supervised and unsupervised learning techniquesBackground in investigating and troubleshooting model performance issuesCapability to map models onto specialized AI hardware through compilation steps and kernelsStrong English communication skills (B2 level or higher)Nice to haveExperience with C++ and kernel programmingKnowledge of GPU/AI-accelerator architecturesWe offerCONTINUOUS UPSKILLING, LEARNING & DEVELOPMENTDiversity of tasks and projectsAssessment center for objective review of competency levelPersonal development planMentoring programs and leadership developmentCertification and professional development supportAccess to learning platforms including more than 2,500 internal coursesEnglish courses taught by certified teachersCORPORATE BENEFITSExtra leave daysReferral bonusesCOMPENSATION PACKAGECompetitive compensation paid in USDRegular salary and performance reviewsMEDICAL & HEALTHCAREPrivate health insuranceWell-being eventsWORKING ENVIRONMENTRecreation areas and kitchensTea, coffee and snacksSports equipment and game consolesIT EquipmentMicrosoft’s Software Assurance Home Use Program (HUP)Please note that our Talent Attraction Team reviews applications and CVs submitted in English.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.