Lead AI Engineer with Machine Learning

Remote • Posted 10 hours ago • Updated 10 hours ago
Full Time
Remote
Fitment

Dice Job Match Score™

🛠️ Calibrating flux capacitors...

Job Details

Skills

  • Team Building
  • Computer Architecture
  • ASIC
  • Neural Network
  • Collaboration
  • Scalability
  • Leadership
  • Python
  • PyTorch
  • Deep Learning
  • Training
  • Machine Learning (ML)
  • Management
  • Unsupervised Learning
  • Computer Hardware
  • English
  • Communication
  • C++
  • Kernel Programming
  • GPU
  • Artificial Intelligence

Summary

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 Requirements 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) Nice to have Experience with C++ and kernel programming Knowledge of GPU/AI-accelerator architectures
Employers have access to artificial intelligence language tools (“AI”) that help generate and enhance job descriptions and AI may have been used to create this description. The position description has been reviewed for accuracy and Dice believes it to correctly reflect the job opportunity.
  • Dice Id: 10330481
  • Position Id: c8b33365d0e5acf5964dd0bd1825ae83
  • Posted 10 hours ago
Create job alert
Set job alertNever miss an opportunity! Create an alert based on the job you applied for.

Similar Jobs

Remote or Huntsville, Alabama

Today

Full-time

Remote or Minnetonka, Minnesota

Today

Full-time

USD 148,900.00 - 255,300.00 per year

Remote or Austin, Texas

Today

Full-time

Remote

Today

Full-time

USD 109,000.00 - 156,000.00 per year

Search all similar jobs