Machine Learning Engineer Jobs at Startups

SpaceXSI is building Super Intelligence to help humanity understand the universe, creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next. You will own and elevate our mission-critical research platform that directly accelerates SpaceXSI's frontier work. The evals team creates and executes instruments for measuring model capabilities and behaviors, used to clean datasets, provide feedback for experiments, and track growth throughout training. We also diagnose why agents fail and how to make improvements, managing execution infrastructure, designing data sets and grading schemes. You will design, build, and ship delightful and bulletproof systems that engineers love to use every day, valuing strong attention to detail, first principles reasoning, and great taste.

$175k - $275k

Mirage is an AI video company focused on making creation dramatically easier by tackling complex challenges in generative media. We are building multimodal agents that transform creative direction into edits, rendering engines that composite video and graphics, and generative video models. Our flagship creative workspace is Captions, powered by our Tesseract engine which provides AI agents with native video-editing capabilities and offers APIs for developers to integrate our models into their products.

$175k - $300k

As a Machine Learning Engineer II – Learned Behaviors, you will help develop and deploy behavior models that power decision-making for autonomous trucks. Working closely with teams across perception, prediction, planning, and safety, you will contribute to learned behavior modules that enable safe, efficient, and human-like driving in real-world freight operations. This role focuses on building, validating, and improving machine learning models and infrastructure that support learned behavior systems within the autonomy stack.

$153k - $184k

As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X. We're looking for exceptional engineers who are passionate about our mission and have a strong desire to make a meaningful impact.

$180k - $440k

The Tracking Team at Torc is responsible for developing the ML models, state estimation algorithms, and production software that enable accurate and highly available information about other actors and objects for the safe operation of autonomous trucks. This role involves designing, developing, and deploying production ML models for multi-object tracking, association, pose and kinematic estimation, and sensor fusion using various vehicle sensors. You will also develop scalable training and evaluation workflows, analyze large-scale vehicle data to improve performance, and develop robust production software in C++ and Python. The position requires making technical design and architecture decisions, collaborating with cross-functional teams, and providing technical leadership.

$177k - $213k

As a Staff Machine Learning Engineer specializing in BEV (Bird's-Eye View) and Multi-Modal Perception, you will lead the development of next-generation models that unify information across cameras, LiDAR, and radar to deliver a rich spatial understanding of the driving environment. You will drive architectural innovation, large-scale model training, and data-driven improvements that directly advance the perception capabilities at the heart of Torc's autonomous driving stack. This is a technical leadership role focused on model innovation and maturity, not downstream feature integration.

$216k - $259k

Torc is seeking a Senior Machine Learning Engineer to join the Ego-Motion & Localization Team. This team is responsible for developing the ML models, state estimation algorithms, and production software that estimate vehicle pose, velocity, and acceleration in real-time. The solutions fuse information from multiple sensors and maintain robust performance under various conditions. As part of this team, you will develop and deploy production-ready localization and sensor fusion solutions, contributing to the challenges of bringing autonomous vehicle technology to commercial applications.

$177k - $213k

Torc is seeking a Machine Learning Engineer II specializing in 3D Perception to join their Multi-Modal Perception team. This role focuses on advancing Bird's Eye View (BEV) perception capabilities for autonomous trucks by developing, evaluating, and improving production machine learning solutions. You will work with a collaborative team to solve complex real-world perception challenges, combining information from various sensors to build robust perception systems that form the foundation for safe and reliable autonomous driving.

$153k - $184k

The Scene Generation team at Torc is at the forefront of developing sensor simulation technology using Neural Rendering and generative models. This technology is crucial for Torc's data-driven simulator, enabling the creation of high-quality data from novel scenarios to address the domain data gap in autonomous driving. The team focuses on state-of-the-art approaches for generating realistic camera, LiDAR, and Radar sensor data, which are vital for training and validation across the entire autonomous vehicle stack. As a Staff ML Engineer, you will play a pivotal role in leading this team, defining its technical direction, and managing daily priorities to advance Torc's AV 3.0 strategy.

$250k - $300k

The Auto Tagger team is responsible for transforming vast amounts of raw, multi-modal vehicle data into a curated library of critical driving scenarios. By identifying long-tail events in driving logs, the team provides essential data to accelerate the development of safe autonomous trucking. Leveraging Pegasus logical layers, this role will structure and catalog findings into an observations database, directly impacting autonomous perception, sensor fusion, and generative simulation testing.

$161k - $193k

Torc is seeking a Senior Machine Learning Engineer focused on 3D Perception to join their Multi-Modal Perception team. This role is crucial for developing the machine learning systems that enable autonomous trucks to perceive and understand their surroundings. You will advance Bird's Eye View (BEV) perception capabilities, creating production-ready solutions to enhance environmental understanding, model robustness, and overall system performance within Torc's autonomy stack. This is a collaborative role where you'll work with engineers and researchers to tackle complex real-world perception challenges.

$177k - $213k

Torc is seeking a Senior ML Engineer to join their Offline Perception team, focusing on creating high-quality, semantically rich training data for end-to-end Vision-Language-Action (VLA) models. This role involves owning the dataset layer that these models learn from, transforming petabytes of logged multi-modal fleet data into VLM/VLA-ready datasets. You will design, implement, and deploy cloud-based pipelines for geometric and semantic annotations, build VLM-assisted auto-labeling systems, and generate reasoning-grounded labels to support model training and explainable driving behavior. The position also requires mining and curating the long tail of rare scenarios, closing the data flywheel by defining schemas and quality metrics, and partnering closely with the end-to-end model team to ensure continuous dataset delivery. You will scale these processes on cloud infrastructure using distributed, reproducible pipelines and disciplined software practices.

$177k - $213k

Torc is seeking a Senior Software Engineer to join our New AI Initiatives team. This role is crucial for building the foundation of Torc's AI development tooling and translating complex business problems into machine learning workflows. You will be responsible for exploratory data analysis, deep learning model development, batch inference, metric computation, and embedded deployment. This position involves working with numerous stakeholders and different teams to ensure Torc embraces state-of-the-art technology, driving exploration and seeing implementations through to production.

$161k - $193k

The Auto Tagger team is central to our data operations, responsible for transforming vast amounts of raw vehicle data into a curated library of critical driving scenarios. By identifying long-tail events within driving logs, this team provides essential data to accelerate the development of safe autonomous trucking. They structure and catalog findings into an observations database, directly supporting advancements in autonomous perception, sensor fusion, and generative simulation testing.

$177k - $213k

Torc is developing software for automated trucks to transform how the world moves freight. As an ML Engineer on the App Engine team, you will be responsible for building the application layer in the truck, creating an SDK/framework for the composition and execution of applications that form the Virtual Driver system. This role is critical to Torc's commercialization efforts and involves developing solutions for complex embedded robotics challenges such as determinism, type safety, test coverage, CUDA, observability, concurrency, and execution graphs. You will collaborate with other engineers to ensure performance and safety in a resource-constrained environment.

$153k - $184k

Torc is seeking a Senior ML Engineer to join the Pseudo-Labeling team, focusing on creating high-quality annotations from sensor data for autonomous vehicle development. This role will own critical challenges in lane line reconstruction and mapping, setting technical direction for lane line accuracy, map creation, and pose accuracy. You will be responsible for designing, implementing, and deploying systems that generate annotations from logged sensor data, leading efforts in lane line network reconstruction, and developing models for map creation and updates. The position also involves owning pose correction for sensor data and diagnosing calibration/sensor errors. You will act as a technical lead on complex projects, mentor other engineers, and stay abreast of the latest AI/ML advancements in autonomous driving and 3D reconstruction, bringing new techniques into production. This role requires independent development of models and algorithms using disciplined software processes, defining data ingestion and governance, and assessing current capabilities for improvement.

$199k - $299k

As a Senior Machine Learning Engineer focused on Camera Models, you will be instrumental in developing and deploying machine learning models that power the camera-based perception systems for autonomous trucks. Your work will involve building and maintaining core vision systems that enable the autonomy stack to understand the environment, detect and localize objects, and estimate scene structure from camera data. You will own the end-to-end development of robust, scalable camera-based models, iterating on system performance and delivering production-ready solutions within the autonomy stack, collaborating closely with perception, data, and infrastructure teams to ensure safe and reliable autonomous driving.

CA$168k - CA$193k

Snorkel AI

Snorkel AI is seeking an early member for its ML & Research Engineering team to help build the data and environments behind high-performing frontier and agentic AI. You will be responsible for making the process of generating and evaluating frontier AI data faster, cheaper, and more rigorous using ML and AI. This role involves studying data generation and evaluation, forming and validating hypotheses with production data, and shipping scalable solutions. You will play a key part in shaping the direction, standards, and growth of the discipline.

$208k - $315k

Moveworks is seeking a Senior Staff ML Engineer to join our AI Safeguards team. In this role, you will be instrumental in designing, building, and shipping critical safeguards within our agent harness. Your work will focus on protecting against prompt injection, single and multi-turn jailbreaks, toxicity, sensitive topics, and sensitive data exposure. You will develop robust evaluation frameworks to measure the efficacy of these safeguards, addressing both false positives and false negatives using diverse datasets. Collaboration with our Security team will be key, translating their findings into actionable eval cases and fixes, and co-developing mitigations. You will also select, integrate, and fine-tune detection models within a shared stack serving both commercial and government environments, while carefully balancing quality, latency, and cost for controls operating in the request path. Additionally, you will build configurable safeguard policies to empower customers in managing sensitive topics and monitor safeguard behavior in production, addressing any regressions.

Salary not listed

MatX is building the compute platform for AGI, developing vertically integrated full-stack solutions from silicon to systems. We focus on hardware and software to train and run the largest ML workloads for AGI, primarily using the Rust programming language. This role involves designing and optimizing kernels that interface directly with our hardware, working in partnership with ML Research and Hardware Engineering teams, and providing expertise on hardware architecture from a programmer's perspective to ensure seamless integration with the software stack.

$160k - $275k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence for home and commercial markets. The Helix team is responsible for developing the core AI systems that power humanoid autonomy. We are seeking a Helix AI Engineer, Video Pretraining to lead the development of large-scale video foundation models trained on diverse real-world and robot-collected data. This role focuses on pretraining models that learn from raw video, capturing motion, interaction, and temporal structure to enable downstream capabilities in perception, prediction, and embodied reasoning.

$200k - $400k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence. These robots are engineered to perform a variety of tasks in both home and commercial markets. We are seeking a Reinforcement Learning Engineer to focus on the development, training, deployment, and evaluation of advanced reinforcement learning algorithms specifically for the whole-body control of our humanoid robots.

$150k - $350k

Sieve

Sieve is a multi-modal lab focused on curating high-quality training datasets across video, audio, images, text, and 3D, aiming to solve the bottleneck in the growth of AI applications. As a perception-focused Machine Learning Engineer, you will build systems that transform raw sensor footage into training data crucial for robotics and world-model teams. This role requires a strong background in computer vision, particularly in shipping perception systems for robots, devices, or autonomous vehicles, and applying these skills to massive datasets.

Salary not listed

OpenSpace is redefining how complex projects are built using an AI-powered Visual Intelligence Platform that leverages computer vision and spatial AI to provide construction teams with a real-time view of on-site activities. This role focuses on owning core algorithms for mapping, localization, image analysis, and 3D reconstruction using real-world construction site data. The position integrates advanced computer vision with modern machine learning methodologies, utilizing classical CV, deep learning (including transformers and multi-modal models), and probabilistic modeling to deliver meaningful customer insights.

CA$168k - CA$220k