Machine Learning Engineer

MAI Agents; Mountain View (CA); On-site; USD 160,000 - 225,000; Full time

Benefits offered by this job

Equity

Health and wellness coverage

401(k) program

Job summary

MAI Agents in Mountain View, CA is looking for a Machine Learning Engineer to architect the ecosystem for AI agents. You'll design the core agentic platform and build scalable systems that significantly impact performance marketing.

Ideal candidates have a Master's or PhD in Computer Science, but those with a Bachelor's and strong programming skills are also encouraged. The role offers a salary of $160,000–$225,000, along with meaningful equity and health coverage.

Qualifications

• Strong proficiency in Python and clean, scalable, maintainable code.
• System-level thinker capable of navigating ambiguity.
• Experience with Large Language Models and agentic frameworks.

Responsibilities

• Design and build the core agentic platform for complex workflows.
• Architect foundational data and signal platforms.
• Collaborate with data scientists to build MLOps infrastructure.

Skills

Python

Machine Learning frameworks (PyTorch, TensorFlow)

MLOps infrastructure (MLflow, Kubernetes)

Data Engineering

Education

Master's or PhD in Computer Science or related field

Bachelor's degree with 2+ years of software engineering experience

Job description

About Us

At MAI (pronounced “my”), we are on a mission to democratize advanced advertising technology. We believe that cutting‑edge marketing tools should be accessible to everyone, not just large enterprises with massive budgets. Our platform uses AI agents to automate and optimize performance marketing, empowering small and mid‑sized businesses to scale their ad spend profitably without the need for an agency or endless hours of manual campaign management.

Founded by ad platform veterans from Google and Instacart, we have raised a $25 million Seed funding round led by Kleiner Perkins to accelerate growth. This capital will be used to expand our product and engineering teams, bringing our vision of intelligent, autonomous marketing to life. Our AI agents have already proven their value, helping clients drive 40% more sales and managing millions in monthly Google Ads spend. Our client waitlist is growing by the day.

Why Join Now

While traditional software has a clear playbook, building the infrastructure for autonomous, intelligent agents is a new frontier — and we’re writing the manual. As an early Machine Learning Engineer at MAI, you will be the architect of the ecosystem where our AI agents live, learn, and operate. You will have a profound impact across our entire stack, from foundational data platforms to core agentic frameworks and scalable serving systems.

You will work on a small, focused team of fewer than ten engineers, with immense impact and the ability to shape the future of our company. We are forming three core teams:

• Product Engineering & Infrastructure
• AI Platform & Foundation
• Agent Application & Quality

What You’ll Do

• Build the Agent’s Operating System: design and build our core agentic platform that orchestrates complex workflows, enables long‑term memory, and integrates human feedback loops.
• Engineer the Data Engine: architect our foundational data and signal platform using a modern lake‑house architecture, building robust pipelines and ML serving systems.
• Create World‑Class Tools for AI: build a suite of secure “tool machines” that allow agents to execute code, manipulate data, and call third‑party APIs.
• Ship an Exceptional Product Experience: develop customer‑facing applications, including a seamless chat UI, and own the scalable serving infrastructure needed for a 24/7 experience.
• Bring Models to Life: collaborate closely with data scientists to build the MLOps infrastructure for training, fine‑tuning, and deploying state‑of‑the‑art reasoning models that define our agents’ intelligence.

What You’ll Bring

• A Master’s or PhD in Computer Science or a related quantitative field, OR a Bachelor’s degree with 2+ years of professional software engineering experience.
• Strong proficiency in Python and a passion for writing clean, scalable, maintainable code.
• A product‑minded mindset that bridges backend systems and delightful user experience.
• A systems‑level thinker capable of navigating ambiguity and designing for scalability, reliability, and extensibility.
• Experience with ML frameworks (e.g., PyTorch, TensorFlow) or MLOps infrastructure (e.g., MLflow, Kubernetes, serving systems).
• Hands‑on experience developing with Large Language Models, agentic frameworks (such as LangGraph), or building RAG systems.

Bonus Points

• Data Engineering: building ETL/streaming pipelines, working with Spark, Airflow, dbt, or lakehouse architectures.

Why You’ll Love Working at MAI

• Unparalleled Learning: solve novel challenges in building scalable, reliable systems for autonomous agents and LLMs.
• High Impact: your work will directly shape our core platform, culture, and customer success.
• A Culture of Curiosity: work with a tight‑knit team that values transparency, first‑principles thinking, and a relentless drive to solve hard problems together.
• True Ownership: significant autonomy over your work and a clear path for growth as the company scales.

Compensation and Benefits

• Salary: $160,000–$225,000, commensurate with experience.
• Equity: meaningful equity to feel invested in our mission.
• Health and Wellness: medical, dental, and vision coverage for you and your family.
• 401(k): competitive program to help you build for the future.

We believe in a holistic approach to hiring. If you’re passionate about our mission and eager to learn and grow, we encourage you to apply even if you don’t meet every requirement. We value potential, curiosity, and hunger. We look forward to hearing from you.