Note: The job is a remote job and is open to candidates in USA. EPAM Systems is a global leader in AI transformation engineering and integrated consulting. The company is seeking a Lead AI Engineer to design, train, optimize, analyze, and deploy neural network models for specialized AI hardware while collaborating with teams focused on compilers, neural networks, LLMs, and CNNs.
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
Skills
- 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)
- Experience with C++ and kernel programming
- Knowledge of GPU/AI-accelerator architectures
Benefits
- Georgia: Opportunity to work abroad for up to two months per year
- Georgia: Relocation opportunities within EPAM offices in 55+ countries
- Georgia: Corporate and social events
- Georgia: Leadership development, career advising, soft skills and well-being programs
- Georgia: Certifications, including GCP, Azure and AWS
- Georgia: Unlimited access to EPAM's internal learning database
- Georgia: Free English classes with certified teachers
- Georgia: Participation in the Employee Stock Purchase Plan
- Georgia: Monetary bonuses for engaging in the referral program
- Georgia: Comprehensive medical and family care package
- Georgia: Five trust days per year (sick leave without a medical certificate)
- Georgia: Benefits package including sports activities and a variety of stores and services
- Georgia: Remote work from anywhere in Georgia, including home, EPAM offices in Tbilisi and Batumi, or a coworking space in Kutaisi
- Kazakhstan: Opportunity to work abroad for up to two months per year
- Kazakhstan: Relocation opportunities within EPAM offices in 55+ countries
- Kazakhstan: Corporate and social events
- Kazakhstan: Leadership development, career advising, soft skills and well-being programs
- Kazakhstan: Certifications, including GCP, Azure and AWS
- Kazakhstan: Unlimited access to EPAM's internal learning database
- Kazakhstan: Free English classes with certified teachers
- Kazakhstan: Discounts in local language schools, including online courses for the Kazakh language
- Kazakhstan: Participation in the Employee Stock Purchase Plan
- Kazakhstan: Monetary bonuses for engaging in the referral program
- Kazakhstan: Medical and family care package
- Kazakhstan: Six trust days per year (sick leave without a medical certificate)
- Kazakhstan: Coverage of psychology sessions of your choice
- Kazakhstan: Benefits package including sports activities and a variety of stores and services
- Kazakhstan: Housing support program with preferential mortgage access via Otbasy Bank partnership
- Kazakhstan: Remote work from home or access to EPAM offices in Astana, Almaty or Karaganda
- Kyrgyzstan: Opportunity to work abroad for up to two months per year
- Kyrgyzstan: Relocation opportunities within EPAM offices in 55+ countries
- Kyrgyzstan: Corporate and social events
- Kyrgyzstan: Leadership development, career advising, soft skills and well-being programs
- Kyrgyzstan: Certifications, including GCP, Azure and AWS
- Kyrgyzstan: Unlimited access to EPAM's internal learning database
- Kyrgyzstan: Free English classes with certified teachers
- Kyrgyzstan: Monetary bonuses for engaging in the referral program
- Kyrgyzstan: Medical and family care package
- Kyrgyzstan: Six trust days per year (sick leave without a medical certificate)
- Kyrgyzstan: Coverage of psychology sessions of your choice
- Kyrgyzstan: Discounts for fitness clubs and sports programs
- Kyrgyzstan: Benefits package including sports activities and a variety of stores and services
- Kyrgyzstan: Remote work from anywhere in Kyrgyzstan, including home or the EPAM office in Bishkek
- Armenia: Opportunity to work abroad for up to two months per year
- Armenia: Relocation opportunities within EPAM offices in 55+ countries
- Armenia: Corporate and social events
- Armenia: Leadership development, career advising, soft skills and well-being programs
- Armenia: Certifications, including GCP, Azure and AWS
- Armenia: Unlimited access to EPAM's internal learning database
- Armenia: Free English classes with certified teachers
- Armenia: Participation in the Employee Stock Purchase Plan
- Armenia: Monetary bonuses for engaging in the referral program
- Armenia: Comprehensive medical and family care package
- Armenia: Four trust days per year for personal needs
- Armenia: Discounts for fitness clubs
- Armenia: Benefits package including hotels, restaurants, stores and services
- Armenia: Fully remote work from any location in Armenia, including home or EPAM offices in Yerevan or Gyumri
- Uzbekistan: Opportunity to work abroad for up to two months per year
- Uzbekistan: Relocation opportunities within EPAM offices in 55+ countries
- Uzbekistan: Corporate and social events
- Uzbekistan: Leadership development, career advising, soft skills and well-being programs
- Uzbekistan: Certifications, including GCP, Azure and AWS
- Uzbekistan: Unlimited access to EPAM's internal learning database
- Uzbekistan: Free English classes with certified teachers
- Uzbekistan: Discounts in local language schools, including offline courses for the Uzbek language
- Uzbekistan: Monetary bonuses for engaging in the referral program
- Uzbekistan: Medical and family care package
- Uzbekistan: Four trust days per year (sick leave without a medical certificate)
- Uzbekistan: Discounts for fitness clubs, dance schools and sports programs
- Uzbekistan: Benefits package including sports activities and a variety of stores and services
- Uzbekistan: Remote work from anywhere in Uzbekistan, including home or the EPAM office in Tashkent
Company Overview
Company H1B Sponsorship