AI Engineer - Machine Learning 2

$124,800 - $124,800/Yr

Mokshaa Consulting LTD - Redmond, WA

Full-time - Mid Level

Hybrid - Redmond, WA

About the position

The AI Engineer - Machine Learning 2 role focuses on advancing the development and improvement of software foundations and tools essential for training state-of-the-art AI models. The position involves creating scalable and efficient training infrastructures that facilitate the entire machine learning process, from data handling to model deployment, particularly in practical robotics scenarios. Collaboration with researchers and software engineers is key to ensuring the smooth integration and functionality of training systems, while also exploring innovative methods for utilizing diverse datasets.

Responsibilities

• Create and uphold efficient, scalable, and distributed training systems including data preprocessing, training orchestration, and model assessment for training 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 and develop benchmarks for evaluating ML 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

• 5+ overall years of experience in the field.
• Minimum 2+ years experience with Linux.
• Minimum 2+ years experience with Python.
• Minimum 2+ years experience with Configuration management.
• 2+ years experience with ROS/ROS2.
• 2+ years experience with TensorFlow.
• 2+ years experience with any sort of machine learning packages.

Nice-to-haves

• 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

• Contract to hire position with a pay rate of $60/hr depending on engagement type and experience.