Lead AI Engineer – Machine Learning & Neural Networks

About the role

We are looking for a Lead AI Engineer to drive the design and deployment of cutting‑edge neural network solutions on specialized AI hardware. You will work with a multidisciplinary team that spans compilers, large language models, and computer architecture, delivering high‑performance AI applications.

Key responsibilities

• Design, implement, and iterate on neural network architectures using PyTorch to meet performance targets.
• Train, validate, and fine‑tune machine‑learning models on diverse datasets, ensuring accuracy and robustness.
• Investigate and resolve model performance issues, continuously improving results.
• Analyze how models map onto Tenstorrent devices through compilation steps and kernel optimization.
• Identify functional and performance gaps, proposing innovative mitigation strategies.
• Collaborate with cross‑functional teams working on compilers, LLMs, CNNs, and hardware design.
• Deploy neural networks for various applications, guaranteeing scalability and reliability.

Required profile

• 5+ years of experience in machine‑learning and AI engineering.
• At least 1 year of leadership or lead‑developer experience.
• Strong proficiency in Python and hands‑on experience with PyTorch.
• Proven expertise in designing, training, and deploying neural networks.
• Solid understanding of supervised and unsupervised learning techniques.
• Experience troubleshooting model performance and mapping models onto specialized AI hardware.
• Excellent English communication skills (B2 level or higher).

Required skills

• Python
• PyTorch
• C++
• Kernel programming
• GPU/AI‑accelerator architectures

What we offer

• Opportunity to work on innovative AI solutions with global impact.
• Flexible remote work from anywhere in Kyrgyzstan (home or modern office in Bishkek).
• Possibility to work abroad up to two months per year and relocation options in 55+ countries.
• Leadership development, career advising, soft‑skill and well‑being programs.
• Certification support for GCP, Azure, and AWS.
• Unlimited access to EPAM’s internal learning platform and free English classes.
• Medical & family care package, referral bonuses, trust days, and psychology session coverage.
• Discounts for fitness clubs, sports programs, and a broad benefits package.