Senior AI/ML Engineer with PyTorch, ++

Back to Search

Senior AI/ML Engineer with PyTorch, ++

Remote in Georgia, & 4 others

AI Solution Engineering

apply

FacebookLinkedInSend via email

Looking for something else?

Find a vacancy that works for you. Send us your CV to receive a personalized offer.

Find me a job

Location-specific conditions & benefits*

Choose an option

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