Remote Jobs RockRemote Jobs Rock

Staff Machine Learning Engineer - Search

📅 Jul 30
Machine LearningSearchEmbeddingsSemantic Search
Apply Now

📜 Description

  • Set the roadmap for search across the platform and prioritize problems to solve.
  • Build embeddings and datasets at scale with data engineers, then deliver semantic search alongside keyword search where appropriate.
  • Measure search quality through feedback loops and improve embeddings and results.
  • Ensure results are relevant to each user and respect access controls.
  • Deliver customer-facing ML features end to end and make build-versus-buy decisions.
  • Lead ML technically, mentor teammates, and influence practices across data and engineering.

🛠️ Requirements

  • Experience building embeddings and using them in production, including choosing between approaches and weighing cost, speed, accuracy and keeping data in house
  • Experience working with large, live datasets (hundreds of terabytes or billions of records), such as data lakes or formats like Parquet or Iceberg
  • A background in machine learning or data engineering, including moving from software engineering
  • Experience building systems that learn from feedback, and measuring whether results are good
  • Experience delivering data or ML products to customers end to end, not only internal stakeholders
  • Experience working technical leadership, guiding a team and influencing across teams
  • Strong relationship building and communication skills, including working across data and software engineering
  • Experience working with data engineers to build datasets at scale
  • Experience building a search engine, even as a proof of concept
  • Experience with OpenSearch, Elasticsearch or vector search tools

✨ Benefits

  • Equity with high growth potential
  • Competitive salary
  • Flexible working arrangements
  • Professional and personal training and development opportunities
  • Hackathons, workshops, and Lunch & Learns
  • Daily breakfast, lunch, and snacks
  • Wellbeing initiatives, including EAP services and a generous parental leave policy
  • On-site gym, table tennis, board games, books library, and pet-friendly offices
Full job description

Why join us?

Mitti by SafetyCulture is a global tech company, just not the kind you're picturing. Our almost 1,000-person team builds tools that make work better for the three billion frontline workers who keep the world moving. We've got big tech scale, without the sign-off layers or corporate theatre. This next chapter is about scaling smart, powered by AI, a clear vision and real ownership.

Big tech impact, without the big tech ick.

About the role
Our mission is to help solve the challenges of an increasingly distributed workforce. As a Staff ML Engineer in the Data team, you’ll shape how we use our data to help customers find what matters and improve. You’ll help set the roadmap for search, build the embeddings and data foundations behind it, and put search on top. We’re at an early stage, with plenty of room to shape how it’s done. The team has the foundations in place and is ready for you to lead the next phase.

In this role, you can expect to:

  • Work out what to solve first, and shape the roadmap for how we apply and grow search across the platform

  • Build and improve the embeddings and datasets that power search, working with data engineers to deliver them at scale

  • Put semantic search on top of those embeddings, combining it with keyword search where it makes sense

  • Build a feedback loop that measures whether search is giving people the right answers, and use it to improve the embeddings and results

  • Make sure the right results reach the right person, respecting access controls and relevance to who is searching and what they do

  • Deliver customer-facing features end to end, from understanding the problem to shipping

  • Act as technical lead for ML in a small team, mentoring others and setting the standard for how we build

  • Build strong relationships across data and engineering teams, and influence how they work without relying on formal authority

  • Weigh effort against impact, including build vs buy decisions

  • Work with our current stack (Databricks, Kafka, Golang, PostgreSQL and OpenSearch on AWS), with room to choose what’s best as the problem becomes clearer

  • Work in a hybrid environment, including at our HQ in Surry Hills, Sydney

  • Demonstrate Mitti values (Think Customer, Open Honest Always, Better as a Team, Be Bold Bring Action)

You may be a great fit if you have:

  • Experience building embeddings and using them in production, including choosing between approaches and weighing cost, speed, accuracy and keeping data in house

  • Experience working with large, live datasets (hundreds of terabytes or billions of records), such as data lakes or formats like Parquet or Iceberg

  • A background in machine learning or data engineering, including moving from software engineering

  • Experience building systems that learn from feedback, and measuring whether results are good

  • Experience delivering data or ML products to customers end to end, not only internal stakeholders

  • Experience working technical leadership, guiding a team and influencing across teams

  • Strong relationship building and communication skills, including working across data and software engineering

  • Experience working with data engineers to build datasets at scale

Nice to have:

  • Experience building a search engine, even as a proof of concept

  • Experience with OpenSearch, Elasticsearch or vector search tools

  • Experience with permission-aware search or personalised results

  • A backend language such as Golang or Node.js

More than a job

  • Equity with high growth potential, and a competitive salary

  • Flexible working arrangements

  • Access to professional and personal training and development opportunities

  • Hackathons, Workshops, Lunch & Learns

  • In-house Culinary Crew serving up daily breakfast, lunch and snacks

  • Barista coffee machine, craft beer on tap, boutique wines and a range of non-alcoholic beverages

  • Wellbeing initiatives such as EAP services and generous parental leave policy

  • Quarterly celebrations and team events

  • On-site gym, table tennis, board games, books library, and pet-friendly offices

We’re committed to building inclusive teams and cultivating a sense of belonging so our people can bring their whole authentic selves to work each day. We seek to make reasonable adjustments throughout our recruitment process to create an even playing field for all candidates. Thanks to the tireless efforts of the entire Mitti team we’ve built an incredible culture which has seen us recognised as a Best Place to Work in Australia , the US and the UK .

Even if you don't meet every requirement listed in the ad, please consider applying for this role. We prioritise inclusion and value individuals with potential over a checklist of qualifications. Don't rule yourself out, hit that apply button if this job resonates with you

You can find out more about life at Mitti via Youtube , Twitter , Instagram and LinkedIn.

To all recruitment agencies, we do not accept resumes or partnership opportunities. Please do not forward resumes to Mitti or any of our employees. We are not responsible for any fees associated with unsolicited resumes.

Wayve

Staff Machine Learning Engineer, Robotics

🕒 11 days ago
Wayve👥 501 - 1000 employees🏢 Computer Software🧠 AI-native

Develop and evaluate machine-learning methods that help autonomous-driving models transfer reliably across vehicle platforms, diagnosing real-world failures and improving production performance.

Machine LearningPytorchPythonRobotics
Plaid

Staff Machine Learning Engineer (Research Scientist) - DFAI

🕒 11 days ago
Plaid👥 10,000+ employees🏢 Software Development

Lead the technical strategy and end-to-end development of foundation models, from data curation and pretraining through production serving, evaluation, and monitoring, while guiding teams across Plaid.

Machine LearningFoundation ModelsTransformersLLMs
Faculty

Lead Machine Learning Engineer (Safety)

🕒 11 days ago
Faculty👥 501 - 1000 employees🏢 Internet

Leads the strategy, architecture, and deployment of secure, scalable machine learning systems, advancing AI safety evaluations and multi-agent tooling while advising clients and mentoring engineering teams.

Machine LearningLarge Language ModelsAI Safety EvaluationMulti-agent Systems
Chime

Staff AI/ML Engineer

🕒 17 days ago
Chime👥 10,000+ employees🏢 Financial Services

Lead end-to-end development of production machine learning systems for cash-flow underwriting, shaping technical direction and turning transaction data into fair, accurate credit decisions and measurable member outcomes.

Machine LearningDeep LearningCredit UnderwritingCash-flow Modeling
Wayve

Principal Machine Learning Engineer, Driving Product

🕒 17 days ago
Wayve👥 501 - 1000 employees🏢 Computer Software🧠 AI-native

Lead hands-on development of the L2 driving model, diagnosing on-road issues and improving performance through data, model, evaluation, and release iterations while shaping a small technical team.

PythonMachine LearningProduction MLData Iteration

Trusted by Remote Workers