Machine Learning Engineer, Federal

Scale Ai - Washington, DC

Full-time - Mid Level

Washington, DC

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

The Machine Learning Engineer at Scale is responsible for integrating advanced techniques in computer vision, deep learning, deep reinforcement learning, and natural language processing into production environments to enhance Scale's products and customer experiences. This role involves leveraging massive datasets to improve machine learning pipelines, particularly for Federal Government clients, and aims to scale operations from millions to billions of tasks monthly.

Responsibilities

• Utilize state-of-the-art models from internal and community sources in production to address customer and tasker challenges.
• Identify areas for improvement in existing models, enhance them through retraining and hyperparameter tuning, and deploy updates without compromising core model characteristics.
• Collaborate with product and research teams to discover improvement opportunities in current and upcoming product lines.
• Work with large datasets to create generic models and fine-tune them for specific products.
• Develop a scalable ML platform to automate machine learning services.
• Act as a representative for applying machine learning techniques across the engineering and product organization.
• Demonstrate the ability to multi-task and quickly learn new technologies.

Requirements

• Extensive experience in computer vision, deep learning, deep reinforcement learning, or natural language processing in a production setting.
• Solid understanding of algorithms, data structures, and object-oriented programming.
• Strong programming skills in Python or Javascript, with experience in TensorFlow or PyTorch.

Nice-to-haves

• Graduate degree in Computer Science, Machine Learning, or Artificial Intelligence specialization.
• Experience with cloud technology stacks (e.g., AWS or GCP) and developing machine learning models in a cloud environment.
• Familiarity with generative AI models.

Benefits

• Comprehensive health, dental, and vision coverage.
• Retirement benefits.
• Learning and development stipend.
• Generous paid time off (PTO).
• Potential commuter stipend.