Senior AI/ML Engineer with PyTorch, С++

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