Haystack
← Back to Jobs
Technology

Senior AI/ML Engineer with PyTorch, ?++

EPAM SystemsUnited States🇺🇸United StatesPosted 31 Jul 2026

Quick Overview

Work Type
Hybrid
Level
Mid Senior

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

Skills

Machine Learning
Deep Learning
C++
PyTorch
Python

Similar jobs