Senior AI/ML Engineer with PyTorch, С++

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