Overview

We are seeking a Lead AI Engineer to join a cutting-edge team building computers designed specifically for AI. Our customer brings together experts in computer architecture, ASIC design, advanced systems, and neural network compilers to build the next generation of computing. The team is actively working on compilers, neural networks, LLMs, and CNNs, and is looking for an experienced engineer to take on a Key Developer role within the ML & AI team.

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, iterating on models and techniques to continuously improve results
  • Analyze how models map onto Tenstorrent devices through compilation steps and kernels
  • Identify gaps in functionality and performance, proposing innovative solutions to mitigate any issues
  • Collaborate with cross-functional teams working on compilers, neural networks, LLMs, and CNNs
  • Deploy neural networks for various applications, ensuring scalability and reliability
Required Qualifications:
  • 5+ years of experience in machine learning and AI engineering
  • At least 1 year of relevant leadership experience
  • Proficiency in Python programming with hands-on experience in PyTorch for developing and training deep learning models
  • Expertise in designing, training, and deploying neural networks for various applications
  • Understanding of machine learning fundamentals, including supervised and unsupervised learning techniques
  • Background in investigating and troubleshooting model performance issues
  • Capability to map models onto specialized AI hardware through compilation steps and kernels
  • Strong English communication skills (B2 level or higher)
Benefits:
  • Delivering innovative solutions to industry leaders, making a global impact
  • Enjoyable working environment, whether it is the vibrant office or the comfort of your home
  • Opportunity to work abroad for up to two months per year
  • Relocation opportunities within our offices in 55+ countries
  • Corporate and social events
  • Leadership development, career advising, soft skills and well-being programs
  • Certifications, including GCP, Azure and AWS
  • Unlimited access to LinkedIn Learning and Udemy
  • Free English classes with certified teachers
  • Participation in the Employee Stock Purchase Plan
  • Monetary bonuses for engaging in the referral program
  • Comprehensive medical & family care package
  • Four trust days per year for personal needs
  • Discounts for fitness clubs
  • Benefits package (hotels, restaurants, stores and services)
Nice To Have:
  • Experience with C++ and kernel programming
  • Knowledge of GPU/AI-accelerator architectures
  • AI Solution Engineering
Note:

✨ Our intelligent job search engine discovered this job and republished it for your convenience.
Please be aware that the job information may be incorrect or incomplete. The job announcement remains the property of its original publisher. To view the original job and its full details, please visit the job's URL on the owner’s page.

Please clearly mention that you have heard of this job opportunity on https://ijob.am.