Lead AI Engineer: Architect Neural Nets on AI Hardware

EPAM Systems Inc; Turkey; On-site; TRY 6,388,000 - 10,319,000; Full time

Benefits offered by this job

Continuous upskilling

Private health insurance

Mentoring programs

Job summary

EPAM Systems Inc. seeks a Lead AI Engineer to join its ML & AI team, focusing on building AI-centric hardware and software solutions.

You will design neural networks in PyTorch, train and fine-tune models, and collaborate across compilers, neural networks, LLMs, and CNNs to deliver scalable AI applications. The role requires 5+ years in ML/AI engineering and at least 1 year in leadership, with strong Python and PyTorch expertise and experience mapping models to AI hardware.

Qualifications

• 5+ years of experience in machine learning and AI engineering.
• At least 1 year of leadership experience.
• Proficiency in Python and PyTorch for developing and training deep learning models.
• Experience mapping models onto specialized AI hardware through compilation steps and kernels.

Responsibilities

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

Skills

Python

PyTorch

Leadership experience

English communication

Tools

C++

Kernel programming

GPU architectures

Job description

EPAM Systems Inc. seeks a Lead AI Engineer to join its ML & AI team, focusing on building AI-centric hardware and software solutions.

You will design neural networks in PyTorch, train and fine-tune models, and collaborate across compilers, neural networks, LLMs, and CNNs to deliver scalable AI applications. The role requires 5+ years in ML/AI engineering and at least 1 year in leadership, with strong Python and PyTorch expertise and experience mapping models to AI hardware.