Lead AI Engineer with Machine Learning

EPAM Systems; Georgia; On-site; USD 160,000 - 240,000; Full time

Benefits offered by this job

Global relocation

Employee stock purchase plan

English language training

Referral bonuses

Medical & family care

Five trust days per year

Company learning resources

Job summary

EPAM Systems Georgia is seeking a Lead AI Engineer to join a cutting-edge team building AI-optimized computers. You will design neural networks, optimize models, and map workloads to Tenstorrent devices, collaborating with experts in architecture, ASICs, and neural network compilers.

This role offers remote work from Georgia, leadership responsibilities, and opportunities to impact next-generation AI hardware across global projects, with strong growth and training programs.

Qualifications

• 5+ years of experience in ML/AI engineering
• At least 1 year of leadership experience
• Strong Python and PyTorch expertise
• Experience deploying neural networks on AI hardware through compilation steps
• Strong English communication (B2+)

Responsibilities

• Design and iterate neural network architectures using PyTorch for diverse tasks
• Train, validate, and fine-tune models on relevant data for high accuracy
• Analyze model performance and iterate to improve results
• Map models to Tenstorrent devices via compilation steps and kernels
• Collaborate with cross-functional teams on compilers, NNs, LLMs, and CNNs
• Deploy neural networks ensuring scalability and reliability

Skills

Python

PyTorch

Leadership

Neural networks

English (B2+)

Tools

C++

Kernel programming

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.

Experience the freedom of remote work from anywhere in Georgia, whether from the comfort of your home, our modern offices in Tbilisi and Batumi or a coworking space in Kutaisi.

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
• Identify gaps in functionality and performance, proposing innovative solutions to mitigate any issues
• Collaborate with cross-functional teams working on compilers, neural networks, LLMs, and CNNs
• Deploy neural networks for various applications, ensuring scalability and reliability

Requirements

• 5+ years of experience in machine learning and AI engineering
• At least 1 year of relevant leadership experience
• Proficiency in Python programming with hands-on experience in PyTorch for developing and training deep learning models
• Expertise in designing, training, and deploying neural networks for various applications
• Understanding of machine learning fundamentals, including supervised and unsupervised learning techniques
• Background in investigating and troubleshooting model performance issues
• Capability to map models onto specialized AI hardware through compilation steps and kernels
• Strong English communication skills (B2 level or higher)

Nice to have

• Experience with C++ and kernel programming
• Knowledge of GPU/AI-accelerator architectures

We offer

• We connect like-minded people
  • Delivering innovative solutions to industry leaders, making a global impact
  • Enjoyable working environment, whether it is the vibrant office or the comfort of your own home
  • Opportunity to work abroad for up to two months per year
  • Relocation opportunities within our offices in 55+ countries
  • Corporate and social events
• We invest in your growth
  • Leadership development, career advising, soft skills and well-being programs
  • Certifications, including GCP, Azure and AWS
  • Unlimited access to EPAM's internal learning database
  • Free English classes with certified teachers
• We cover it all
  • Participation in the Employee Stock Purchase Plan
  • Monetary bonuses for engaging in the referral program
  • Comprehensive medical & family care package
  • Five trust days per year (sick leave without a medical certificate)
  • Benefits package (sports activities, a variety of stores and services)

EPAM Georgia is a team of innovators united by a passion for technology. The dynamic and inclusive culture we embrace helps positively impact our communities, clients, and employees. Here you will collaborate with multi-national teams, contribute to numerous cutting-edge projects, deliver the most creative solutions, and have an opportunity to learn. Our people are at the heart of our success, and we are proud to provide talents with a solid ground to develop and grow.