Machine Learning Data Engineer Jobs in California

Note: We are currently recruiting for multiple positions, however please only apply for the role that best aligns with your skillset and career goals.

As a Machine Learning Engineer in Delivery, you are a problem solver who stays anchored to impact. You are someone who can grasp advanced engineering concepts across multiple industries, and excel at working directly with customers (and often side-by-side with them on-site) to embed cutting-edge AI models into tools that are useful and used.
You've shipped ML systems end-to-end and at scale: you design, build and test reliable, scalable ML data pipelines; you know how to explore and manipulate 3D point-cloud and mesh data to enable geometry-aware modelling; you select the right libraries, frameworks and tools. Working at the intersection of data science and software engineering, you translate R&D and project outputs into reusable libraries, tooling and products.
With at least 2 years industry experience (post Masters or PhD) in a commercial, non-research environment. You're truly excited about taking ownership of complex work streams and guiding teams to success, while continuously improving the systems and solutions you work on to ensure they are practical, impactful and meet the evolving needs of our customers.

We have hybrid offices in London, New York, and Singapore; this role is hybrid based in the San Francisco area.

As a Machine Learning Engineer, you'll work closely with our Data Scientists, Simulation Engineers, and customers to understand and define the engineering and physics challenges we are solving. You will iterate with customers and use your influence to drive decisions around reliable deployment with measurable outcomes.

• Work closely with our simulation engineers, data scientists and customers to develop an understanding of the physics and engineering challenges we are solving
• Design, build and test data pipelines for machine learning that are reliable, scalable and easily deployable
• Explore and manipulate 3D point cloud & mesh data
• Own the delivery of technical workstreams
• Create analytics environments and resources in the cloud or on premise, spanning data engineering and science
• Identify the best libraries, frameworks and tools for a given task, make product design decisions to set us up for success
• Work at the intersection of data science and software engineering to translate the results of our R&D and projects into re-usable libraries, tooling and products
• Continuously apply and improve engineering best practices and standards and coach your colleagues in their adoption

• Experience applying Machine learning methods (including 3D graph/point cloud deep learning methods) to real-world engineering applications, with a focus on driving measurable impact in industry settings.
• Experience in ML/Computational statistics/Modelling use-cases in industrial settings (for example supply chain optimisation or manufacturing processes) is encouraged.
• A track record of scoping and delivering projects in a customer facing role
• 2+ years' experience in a data-driven role, with exposure to software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps)
• Building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., TensorFlow, MLFlow)
• Distributed computing frameworks (e.g., Spark, Dask)
• Cloud platforms (e.g., AWS, Azure, GCP) and HP computing
• Containerization and orchestration (Docker, Kubernetes)   
• Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly
• Excellent collaboration and communication skills - with teams and customers alike
• A background in Physics, Engineering, or equivalent

Help shape an AI-native engineering company at a formative stage, tackling problems that genuinely matter for industry and society. This is work with real-world impact - and something you can be proud to stand behind.

Work with a high-caliber, collaborative team of engineers, scientists, and operators who care deeply about doing great work, and about helping each other get better. We come from diverse backgrounds, but we share a commitment to operating at the highest level and addressing some of the most complex challenges out there. If you're ambitious, thoughtful, and driven by impact, you'll feel at home.

We operate with a flat structure: good ideas win - wherever they come from. Questioning assumptions and challenging the status quo isn't just welcomed, it's expected.

And it doesn't stop there ...

 Equity options - share meaningfully in the company you're helping to build.

 5% contribution to 401(k) - build long-term security with a strong retirement plan.

 Private health insurance - comprehensive cover for you, offering total peace of mind.

 Enhanced parental leave - 3 months full pay paternity and 6 months full pay maternity leave, to provide extra flexibility during the moments that matter most.

 20 days of Annual Leave (+ Public Holidays) - because taking time to rest matters.

 Personal development - dedicated support for learning, development, and leveling up over time.

 Gympass / Wellhub (subsidized) - for you and up to 3 family members, supporting both physical and mental wellbeing.

 Flexible Spending Account (FSA) - set aside pre-tax dollars for eligible healthcare expenses.

Watch this space, we're continuing to build this as we grow...

$150,000 - $190,000 depending on experience 
Seniority will be assessed throughout our interview process