Staff/Principal Machine Learning Engineer

arrow

San Francisco / $250000 - $350000 annum

INFO

Salary

SALARY:

$250000 - $350000

Location

LOCATION

San Francisco

Job Type
JOB TYPE

Permanent

About the Company

We are a rapidly growing technology organization focused on helping enterprises unlock greater value from their data through advanced artificial intelligence and machine learning solutions. Our platform empowers organizations to transform complex data assets into actionable insights by combining intelligent automation, modern data engineering practices, and intuitive user experiences. We work with some of the most sophisticated data environments in the world and are building the next generation of AI-powered enterprise software.

The Opportunity

As an AI/ML Engineer, you will play a key role in designing and delivering intelligent systems that power advanced automation, knowledge retrieval, code generation, and decision-support capabilities. This position offers significant ownership and influence over technical direction, architecture decisions, and the development of scalable AI solutions.

You will collaborate closely with engineering, product, and platform teams to create AI-driven experiences that solve complex business and technical challenges. The level and scope of responsibility will be aligned to experience, with opportunities for substantial impact at every level.

Key Responsibilities

AI Agent Architecture & Optimization

  • Design, develop, and enhance multi-step AI agents capable of reasoning, planning, and executing complex workflows.
  • Define strategies for context management, task decomposition, error handling, and response quality.
  • Establish evaluation frameworks and performance benchmarks to continuously improve agent effectiveness.
  • Drive experimentation and innovation across emerging AI technologies and methodologies.

Knowledge Retrieval & Information Systems

  • Design and maintain retrieval-augmented generation (RAG) pipelines and supporting infrastructure.
  • Develop solutions leveraging vector search, knowledge graphs, semantic search, and related technologies.
  • Create knowledge representations that enable accurate reasoning across structured and unstructured enterprise data.
  • Improve information retrieval quality, relevance, and scalability.

AI-Powered Code Generation & Data Processing

  • Build and optimize systems that convert natural language requests into reliable technical outputs.
  • Develop prompt engineering strategies, model selection approaches, and evaluation methodologies.
  • Implement safeguards, validation mechanisms, and quality controls for AI-generated outputs.
  • Support intelligent automation across data transformation, analytics, and engineering workflows.

Production-Scale AI Systems

  • Deploy, monitor, and optimize AI solutions operating in large-scale production environments.
  • Improve latency, throughput, reliability, observability, and operational efficiency.
  • Evaluate tradeoffs between model optimization, infrastructure improvements, and system architecture changes.
  • Contribute to best practices for AI governance, monitoring, and lifecycle management.

Qualifications

What We're Looking For

  • Proven experience developing and supporting AI/ML systems in production environments.
  • Strong understanding of evaluation methodologies, performance measurement, and failure analysis.
  • Demonstrated ability to make architectural decisions in ambiguous or rapidly evolving technical environments.
  • Passion for learning new technologies and adapting to emerging industry trends.
  • Strong communication skills with the ability to influence technical direction and participate in engineering design discussions.
  • Experience balancing rapid execution with long-term system scalability and maintainability.

Experience

  • Approximately 4+ years of professional software engineering experience with meaningful exposure to machine learning, artificial intelligence, or related technologies.
  • Candidates ranging from experienced individual contributors to senior technical leaders are encouraged to apply.

Preferred Technical Background

Core Technologies

  • Strong programming skills in Python.
  • Hands-on experience with large language models (LLMs), generative AI applications, and agent-based systems.
  • Experience building retrieval systems using vector databases, semantic search technologies, and/or knowledge graphs.
  • Familiarity with cloud-native architectures and distributed systems.

Nice-to-Have Experience

  • Knowledge of Kubernetes and modern cloud platforms.
  • Experience with Scala, Go, or other systems-level languages.
  • Experience working with modern analytics and data processing platforms.
  • Exposure to technologies such as Apache Spark, cloud data warehouses, data lakes, and distributed computing frameworks.
  • Background in data engineering, business intelligence, analytics, or large-scale data processing environments.

Why Join Us?

  • Opportunity to work on cutting-edge AI and machine learning challenges.
  • Meaningful ownership and influence over technical direction and product innovation.
  • Collaborative environment with highly talented engineers and technical leaders.
  • Professional growth opportunities and support for continued learning.
  • Competitive compensation package, equity opportunities, and comprehensive benefits.
  • Flexible work arrangements and a culture that values impact, innovation, and execution.

Equal Opportunity Employer

We are committed to creating a diverse and inclusive workplace. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, age, disability, veteran status, sexual orientation, gender identity, or any other protected characteristic under applicable law.


CONTACT

Michael DeVita

Recruitment Consultant

SIMILAR
JOB RESULTS

4k-Harnham_DA copy
CAN’T FIND THE RIGHT OPPORTUNITY?

STILL
LOOKING?

If you can’t see what you’re looking for right now, send us your CV anyway – we’re always getting fresh new roles through the door.