Lead AI Engineer with Machine Learning

EPAM Systems is seeking a Lead AI Engineer to join a team building computing systems designed specifically for artificial intelligence. The role focuses on developing, training, optimizing, compiling, and deploying neural network models while collaborating with teams working on compilers, LLMs, CNNs, and specialized AI hardware.

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)
• Experience with C++ and kernel programming
• Knowledge of GPU/AI-accelerator architectures