AI Engineer - Machine Learning

Talent Software Services - Redmond, WA

Full-time - Entry Level

Redmond, WA

Administrative and Support Services

About the position

The AI Engineer - Machine Learning position at Talent Software Services focuses on developing and enhancing training infrastructures for state-of-the-art AI models, particularly in robotics and reinforcement learning. The role involves collaboration with researchers and software engineers to ensure efficient integration and functionality of training systems, with a strong emphasis on scalability and performance improvement.

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 Client 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

• 2-5 years of overall experience in the field, with a preference for 5+ years if possible.
• A Bachelor's degree in computer science or a related field is required.
• Minimum 2+ years experience with Linux.
• Minimum 2+ years of experience with Python.
• Minimum 2+ years experience with Configuration management.
• 2+ years of experience with ROS/ROS2.
• 2+ years of experience with TensorFlow.
• 2+ years of 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 position with a strong chance for long-term extension.