AI Engineer - Machine Learning 2

OSI Engineering - Seattle, WA

Full-time - Entry Level

Seattle, WA

Administrative and Support Services

About the position

Join our team as a Machine Learning Engineer and play a pivotal role in advancing our software foundation and tools essential for training cutting-edge AI models. You'll develop scalable and efficient training infrastructures, collaborate with researchers and engineers, and push the boundaries of AI in practical robotics applications.

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 use of resources, 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 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

• 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

• Hourly Rate Range: $85-100/h
• Hybrid work schedule
• Contract duration of 7.5 months, with a chance of extension