Machine Learning Engineer

$160,618 - $178,464/Yr

Aquent - Seattle, WA

Full-time - Mid Level

Seattle, WA

Administrative and Support Services

About the position

The Machine Learning Engineer will focus on advancing the development and improvement of software tools essential for training state-of-the-art AI models. This role involves creating scalable and efficient training infrastructures that support the entire machine learning process, from data handling to model deployment, particularly in practical robotics applications. The engineer will collaborate with researchers and software engineers to ensure seamless integration and functionality of training systems, while also exploring innovative methods to utilize diverse datasets effectively.

Responsibilities

• Create and uphold efficient, scalable, and distributed training systems, including data preprocessing, training orchestration, and model assessment for large-scale AI models.
• Enhance the efficiency of training procedures to improve performance and resource utilization while maintaining scalability and dependability.
• Collaborate with researchers to create training and evaluation pipelines for state-of-the-art algorithms.
• Develop and design benchmarks for evaluating machine learning models.
• Perform training and fine-tuning of foundation models for robotic applications.
• Monitor and analyze pipelines, identifying bottlenecks and proposing solutions to improve efficiency and performance.
• Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration.

Requirements

• BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience.
• Proficiency in Python, C++, or similar and at least one deep learning library such as PyTorch, TensorFlow, JAX, etc.
• Strong background in distributed computing, parallel processing techniques, handling large-scale datasets and data preprocessing.
• Deep understanding of state-of-the-art machine learning techniques and models.
• Experience with cloud-based training environments (AWS, Google Cloud, Azure).
• Experience in developing and maintaining software tooling and infrastructure for machine learning.
• Deep understanding and practical experience with software engineering principles, including algorithms, data structures, and system design.
• Experience with continuous integration and automated testing frameworks.

Benefits

• Subsidized health, vision, and dental plans
• Paid sick leave
• Retirement plans with a match
• Free online training through Aquent Gymnasium