Machine Learning Infra Engineer

Location: New York, NY (Flatiron District)

Employment Type: Full-time

On-site: 5 days/week in-person (Non-negotiable)

About the Role

Our client is a high-growth, venture-backed defense tech company working at the cutting edge of signal processing, machine learning, and embedded systems. As a Machine Learning Infra Engineer , you will design and build core infrastructure from scratch to handle petabytes of radio spectrum data. This is a high-ownership role for an experienced engineer who has strong opinions on tooling best practices and can make informed, future-proof decisions about architecture.

This team operates with a high bar for talent, a culture of high growth and high ownership , and requires an in-person presence to foster collaboration.

What You'll Do

• Scale distributed data storage and write Python APIs to make loading massive datasets (e.g., 30GB) feel instantaneous.
• Set up orchestration for model training on GPU clusters, including versioning and artifact deployment.
• Explore creative ways to combine relational and vector-based search queries to enable researchers to quickly discover the most relevant data.
• Define large areas of the engineering roadmap and collaborate extensively with researchers.

What Our Client is Looking For

• 2-7 years of experience in ML Ops, building platforms that track, package, and ship models in production.
• Experience designing and implementing data infrastructure from scratch (databases, cloud storage, cloud compute).
• Strong experience with AWS , including networking, S3, Sagemaker, RDS, ECS, Lambda, and related infra-as-code tools (AWS CDK).
• Experience managing a production-grade Python codebase used by other people.
• Ability to deal with larger-than-memory data inexpensively without setting up a cluster.
• Experience with ML experiment tracking, model versioning, and artifact deployment (e.g., MLflow).
• U.S. Citizenship is required due to the nature of their work (Defense/ITAR).

• Salary: $130K - $210K
• Equity: 0.05% - 0.2%
• Visa sponsorship is not available (U.S. citizens only).
• Relocation support is offered for strong candidates moving to NYC.

Due to the high volume of applications we anticipate, we regret that we are unable to provide individual feedback to all candidates. If you do not hear back from us within 4 weeks of your application, please assume that you have not been successful on this occasion. We genuinely appreciate your interest and wish you the best in your job search.

Commitment to Equality and Accessibility:

At MLabs, we are committed to offer equal opportunities to all candidates. We ensure no discrimination, accessible job adverts, and providing information in accessible formats. Our goal is to foster a diverse, inclusive workplace with equal opportunities for all. If you need any reasonable adjustments during any part of the hiring process or you would like to see the job-advert in an accessible format please let us know at the earliest opportunity by emailing [email protected].

MLabs Ltd collects and processes the personal information you provide such as your contact details, work history, resume, and other relevant data for recruitment purposes only. This information is managed securely in accordance with MLabs Ltd’s Privacy Policy and Information Security Policy, and in compliance with applicable data protection laws. Your data may be shared only with clients and trusted partners where necessary for recruitment purposes. You may request the deletion of your data or withdraw your consent at any time by contacting [email protected].