Machine Learning Engineer AI Models

You will port, tune, validate, and debug LLM and vision workloads on a custom AI accelerator. You will analyze model behavior and bottlenecks across the stack, improve execution efficiency and correctness, and translate advanced model architectures into reliable production deployments.

Responsibilities

• Bring up, run, and debug modern machine learning models using PyTorch or TensorFlow
• Analyze model behavior and performance to identify bottlenecks
• Improve the efficiency, correctness, and scalability of model execution
• Collaborate with compiler, kernel, and hardware specialists on system-level improvements
• Translate advanced model architectures into high-performance production deployments

Requirements

• Experience building and working with machine learning models in PyTorch or TensorFlow
• Understanding of modern machine learning model architectures such as transformers
• Software engineering fundamentals
• Debugging skills
• Problem-solving skills