Senior Machine Learning Engineer , AGI Structured Information Retrieval

Amazon - Boston, MA

Full-time - Senior

Boston, MA

Sporting Goods, Hobby, Musical Instrument, Book, and Miscellaneous Retailers

About the position

The Senior Machine Learning Engineer (MLE) will lead the development of algorithms and modeling techniques to enhance the Amazon Knowledge Graph (AKG) and improve natural language to structured retrieval services. This role involves collaborating with a team of engineers and applied scientists to create and scale machine learning models, ensuring high precision and low latency access to knowledge.

Responsibilities

• Quickly learn cutting-edge technologies and algorithms in the field of ML.
• Develop and maintain key platforms for developing, evaluating, and deploying ML models for real-world applications.
• Investigate design approaches, prototype new technology, and evaluate technical feasibility with team members.
• Work closely with applied scientists to develop and scale machine learning models.

Requirements

• 5+ years of professional software development experience in distributed systems with an emphasis on ML infrastructure.
• 5+ years of programming experience building ML infrastructure using languages such as Python, C++, or Rust.
• Hands-on experience with parallel computing platforms such as CUDA, OpenMP, etc.
• Deep understanding of AI frameworks such as PyTorch, TensorFlow, and JAX, and their demands on underlying compute infrastructure.
• Knowledge of emerging AI hardware accelerators and architectures.
• Experience with containerization and orchestration technologies (Docker, Kubernetes).

Nice-to-haves

• 5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• Bachelor's degree in computer science or equivalent.

Benefits

• Diverse and inclusive workplace
• Equal opportunity employer