Senior AI/ML Engineer with PyTorch, С++

EPAM Systems; Georgia; On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Relocation opportunities

Work abroad up to 2 months/year

Corporate and social events

Employee stock purchase plan

Free English classes

Learning database access

Job summary

EPAM Systems Georgia is seeking a Senior AI/ML Engineer to drive design, training, and deployment of advanced deep learning models. You will work remotely from Georgia, collaborating with our ML & AI team across diverse tasks and environments.

The role emphasizes PyTorch, neural networks, and robust ML fundamentals, with strong emphasis on deployment, performance tuning, and cross‑functional collaboration. Relocation opportunities and global collaboration are available.

Qualifications

• 3+ years of Python development with PyTorch for DL models.
• Experience designing, training, and deploying neural networks.
• Strong ML fundamentals in supervised and unsupervised learning.
• Ability to troubleshoot and optimize model performance.
• English proficiency at B2 level or higher.

Responsibilities

• Design, implement, and iterate on neural networks using PyTorch.
• Train, validate, and fine-tune models on diverse datasets.
• Troubleshoot performance issues and iterate on methods.
• Analyze model mapping to devices through compilation steps.
• Deploy neural networks across environments and use cases.
• Apply ML fundamentals to supervised and unsupervised learning outcomes.

Skills

Python

PyTorch

Neural networks

ML fundamentals

Troubleshooting

English proficiency

Tools

C++

Kernel programming

GPU architectures

Job description

We are looking for a Senior AI/ML Engineer to join the ML & AI team, driving the design, training, and deployment of advanced deep learning models.

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, and iterate on models and techniques to continuously improve results
• Analyze how models map onto the client's devices through compilation steps and kernels
• Identify gaps in functionality and performance, and propose innovative solutions to mitigate any issues
• Deploy neural networks for various applications across different environments
• Apply machine learning fundamentals to ensure effective supervised and unsupervised learning outcomes

Requirements

• 3+ years of experience in Python programming, with hands‑on expertise in PyTorch for developing and training deep learning models
• Proficiency in designing, training, and deploying neural networks for various applications
• Understanding of machine learning fundamentals, including supervised and unsupervised learning techniques
• Expertise in troubleshooting and optimizing model performance across diverse tasks and datasets
• English proficiency at B2 level or higher

Nice to have

• Familiarity 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.