Source Job

US

  • Develop and train ML models for learned behavior systems using imitation and reinforcement learning.
  • Write production-quality ML code for training, evaluation, and inference in the autonomy stack.
  • Collaborate with teams to test and integrate learned behavior models across diverse driving environments.

Python PyTorch Machine Learning Reinforcement Learning

13 jobs similar to Machine Learning Engineer II - Learned Planning (Reinforcement Learning)

Jobs ranked by similarity.

US

  • Design high-performance training platform components including orchestration, observability, and performance tuning.
  • Deliver end-to-end ML pipelines from dataset curation to training, validation, and deployment.
  • Drive technical direction across ML Platform, Infrastructure, Autonomy, and Safety teams.

Stack is developing revolutionary AI and advanced autonomous systems for safer, more reliable, and efficient operations, focusing on autonomous trucking. The Stack team brings decades of experience in real-world systems and is committed to a culture of inclusion, entrepreneurship, and innovation.

$225,000–$250,000/yr
US

  • Design and implement state-of-the-art ML models and training pipelines for robotics.
  • Develop efficient data/training strategies and evaluation frameworks for rapid experimentation.
  • Collaborate with engineering to optimize training infrastructure and deployment.

We're revolutionizing real-world automation by making robotic systems accessible to everyone. Our AI-powered platform brings software automation to physical spaces, and we're a small startup team working across the stack to solve customer problems.

US

  • Develop advanced ML models and agentic workflows to accelerate model development.
  • Use AI-assisted tools like Claude and Cursor to investigate model behavior and automate analysis.
  • Set technical direction, mentor engineers, and raise the bar for modeling rigor.

Reddit is a community of communities, built on shared interests, passion, and trust, and is home to the most open and authentic conversations on the internet. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information.

US Unlimited PTO

  • Build the component layer around our layout synthesis engine, including API contracts, services, and evaluation gates.
  • Turn model retraining into a one-command job with built-in benchmarks and readable results for the whole team.
  • Own latency and cost budgets for learned capabilities and partner with infrastructure engineers on MLOps and deployment.

Higharc is a VC-backed startup that is changing how new homes are designed and built using spatial AI and generative floor plan technology. The company is fully remote, has raised over $175M, and values flexibility, collaboration, and asynchronous deep work.

$170,170–$286,000/yr
North America

  • Design and maintain reliable, low-latency ML APIs to integrate Safety AI model outputs into cloud applications.
  • Build scalable data pipelines for continuous model iteration, backtesting, and online evaluation.
  • Optimize model artifacts for production and monitor rollout health, ensuring predictable failure modes.

Samsara builds a Connected Operations Cloud that helps physical operations use IoT data to improve safety, efficiency, and sustainability. Samsara is a recently public company with an employee-led remote culture and a long-term focus.

$230,000–$322,000/yr
US

  • Lead the development and optimization of machine learning models to detect and prevent AI security risks like prompt injection and jailbreaks.
  • Build reproducible training and evaluation pipelines on Reddit's ML platform, partnering with platform engineers to improve performance and reliability.
  • Set the technical vision and multi-quarter modeling roadmap, mentoring engineers and establishing best practices for responsible ML development.

Reddit is a community of communities, built on shared interests and authentic conversations, with 100,000+ active communities and 130 million daily active visitors. It is one of the internet's largest sources of information, fostering a culture of openness and trust.

India

  • Build reinforcement learning and agent environments for real customer and Frontier lab use cases, including task specifications, scoring, and evaluation.
  • Develop benchmarks and evaluation harnesses to measure model and data quality across accuracy, robustness, safety, latency, and cost.
  • Run fine-tuning, adapter, and other model experiments to evaluate how data and methods influence model behavior, and deploy local or self-hosted models for evaluation and inference.

Appen has been a leader in AI training data for over 30 years, specializing in human-generated data to train, fine-tune, and evaluate models across generative AI, LLMs, computer vision, and speech recognition. They support model development through an AI-assisted data annotation platform and a global crowd of over 1 million contributors in more than 200 countries, fostering a culture of innovation, collaboration, and humility over ego.

United States

  • Design and train advanced ML models (e.g., DNNs, transformers) to power Reddit Ads Ranking.
  • Develop and optimize features including embeddings, contextual signals, and cross-session behavior.
  • Mentor other MLEs and contribute to modeling best practices across the organization.

Reddit is a community of communities built on shared interests, passion, and trust. With over 100,000 active communities and approximately 130 million daily active unique visitors, it is one of the internet's largest sources of information and employs a flexible first workforce.

$215,250–$285,982/yr
US

  • Build cutting-edge technology using deep learning and machine learning to personalize Pinterest.
  • Partner with teams to improve ML models for product surfaces like Homefeed and Search.
  • Use data-driven methods to improve content recommendation and ads delivery.

Pinterest is a platform where millions of people find creative ideas and plan for memories. The company is mission-driven, focusing on innovation and inclusion, with a large and diverse user base.

US Unlimited PTO

  • Build the foundations of the EdgeRunner Research organization, including data pipelines, model evaluation, and efficient parallelization.
  • Own codebases in areas like distributed training, quantization, compression, or compute cluster management.
  • Collaborate with a team of self-starters who operate independently and translate business objectives into technical solutions.

EdgeRunner AI builds state-of-the-art AI models for the tactical edge, enabling warfighters to make faster decisions and interact with robotic platforms. As a Series-A startup, we move quickly, fostering a culture of initiative, ownership, and comfort with ambiguity.

$180,000–$270,000/yr
US

  • Build and operate production ML systems for recommendations, ranking, and personalized action selection across Stripe's Growth Platform.
  • Improve contextual bandit and policy-learning approaches, including exploration, reward design, and adaptive user feedback.
  • Develop reliable data and feature pipelines, reusable tooling, and online experiments to drive measurable business outcomes.

Stripe is a financial infrastructure platform for businesses of all sizes, enabling payments and revenue growth. With millions of companies as customers, Stripe fosters a culture of passion, grit, and integrity while working to increase the GDP of the internet.

US Canada Unlimited PTO

  • Design, build, and ship custom internal AI tooling, agents, and workflows for autonomy and research.
  • Integrate and extend third-party AI tools and evaluate new AI models with structured pilots.
  • Partner with cross-functional teams to prototype solutions and drive adoption of AI tools.

Waabi is a leader in Physical AI, founded by AI visionary Raquel Urtasun. The company is growing quickly with offices in Toronto, San Francisco, Dallas, and Pittsburgh, and seeks diverse, innovative candidates.

$189,507–$274,604/yr
Global

  • Develop and improve ads ranking models, including prediction objectives, feature interactions, user-history modeling, and calibration.
  • Take end-to-end ownership of machine learning systems from data pipelines to production integration.
  • Evaluate and apply advances in deep learning and recommendation modeling to improve ads ranking within production constraints.

Quora operates two knowledge-sharing platforms: Quora, a global Q&A platform, and Poe, a platform for interacting with AI language models. They have a remote-first culture with passionate, collaborative, and high-performing global teams focused on transparency and experimentation.