Lead AI Engineer with Machine Learning

EPAM Systems; Turkey; On-site; TRY 5,747,000 - 8,621,000; Full time

Benefits offered by this job

Continuous upskilling

Private health insurance

Learning platforms access

Mentoring programs

Job summary

EPAM is seeking a Lead AI Engineer to join a cutting-edge team building AI-specific computing systems. You will design and deploy neural networks on specialized hardware, collaborating with experts in architecture, ASIC design, and neural network compilers to push the boundaries of AI inference.

Responsibilities include developing models with PyTorch, training and tuning on diverse data, mapping models to Tenstorrent devices via compilation steps, and leading a small ML&AI team to deliver

Qualifications

• 5+ years of experience in machine learning and AI engineering.
• At least 1 year of leadership experience.
• Proficiency in Python with hands-on PyTorch experience.
• Experience designing, training, and deploying neural networks.

Responsibilities

• Design, implement, and iterate neural network architectures using PyTorch for performance on diverse tasks.
• Train, validate, and fine-tune models on relevant datasets for high accuracy and robustness.
• Troubleshoot model performance issues and iterate on techniques to improve results.
• Map models onto Tenstorrent devices through compilation steps and kernels.
• Collaborate with cross-functional teams on compilers, neural networks, LLMs, and CNNs.
• Deploy neural networks for scalable, reliable applications.

Skills

PyTorch

Python

Leadership experience

ML model deployment

Tools

Tenstorrent hardware

GPU kernels

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.

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

• CONTINUOUS UPSKILLING, LEARNING & DEVELOPMENT
  • Diversity of tasks and projects
  • Assessment center for objective review of competency level
  • Personal development plan
  • Mentoring programs and leadership development
  • Certification and professional development support
  • Access to learning platforms including more than 2,500 internal courses
  • English courses taught by certified teachers
• CORPORATE BENEFITS
  • Extra leave days
  • Referral bonuses
• COMPENSATION PACKAGE
  • Competitive compensation paid in USD
  • Regular salary and performance reviews
• MEDICAL & HEALTHCARE
  • Private health insurance
  • Well-being events
• WORKING ENVIRONMENT
  • Recreation areas and kitchens
  • Tea, coffee and snacks
  • Sports equipment and game consoles
  • IT Equipment
  • Microsoft's Software Assurance Home Use Program (HUP)

Please note that our Talent Attraction Team reviews applications and CVs submitted in English.

EPAM is global leader in AI transformation engineering and integrated consulting, serving Forbes Global 2000 companies and ambitious startups. With over thirty years of expertise in custom software, product and platform engineering, we empower our clients to become AI-Native enterprises, driving measurable value from innovation and digital investments.