Skip to content
EmaEmaMountain View, CA

Software Engineer, Machine Learning

Develops and deploys ML models for NLP, retrieval, ranking, reasoning, dialog, and code-generation systems. Requires Master's/PhD, 2+ years experience with production ML, deep NLP expertise, Python, and frameworks like PyTorch/TensorFlow.

135k – 200k/yr
Hybrid2+ YOEML Engineering

About the role

Responsibilities

  • Conceptualize, develop, and deploy machine learning models that underpin our NLP, retrieval, ranking, reasoning, dialog and code-generation systems.
  • Implement advanced machine learning algorithms, such as Transformer-based models, reinforcement learning, ensemble learning, and agent-based systems to continually improve the performance of our AI systems.
  • Process and analyze large, complex datasets (structured, semi-structured, and unstructured), and use your findings to inform the development of our models.
  • Work across the complete lifecycle of ML model development, including problem definition, data exploration, feature engineering, model training, validation, and deployment.
  • Implement A/B testing and other statistical methods to validate the effectiveness of models. Ensure the integrity and robustness of ML solutions by developing automated testing and validation processes.
  • Clearly communicate the technical workings and benefits of ML models to both technical and non-technical stakeholders, facilitating understanding and adoption.

Minimum Qualifications

  • A Master's degree or Ph.D. in Computer Science, Machine Learning, or a related quantitative field.
  • At least 2 years of industry experience in building and deploying production-level machine learning models.
  • Deep understanding and practical experience with NLP techniques and frameworks, including training and inference of large language models.
  • Deep understanding of any of retrieval, ranking, reinforcement learning, and agent-based systems and experience in how to build them for large systems.
  • Proficiency in Python and experience with ML libraries such as TensorFlow or PyTorch.

Ideally, You'd Have

  • Excellent skills in data processing (SQL, ETL, data warehousing) and experience working with large-scale data systems.
  • Experience with machine learning model lifecycle management tools, and an understanding of MLOps principles and best practices.
  • Familiarity with cloud platforms like GCP or Azure.
  • Familiarity with the latest industry and academic trends in machine learning and AI, and the ability to apply this knowledge to practical projects.
  • Good understanding of software development principles, data structures, and algorithms.

Compensation (California Based)

  • Standard base salary: $135,000 to $200,000 annually.
  • Compensation determined by factors such as location, level, job-related knowledge, skills, and experience. Certain roles may be eligible for variable compensation, equity, and benefits.

Skills

PythonPyTorchTensorFlowNLPTransformersReinforcement LearningRetrievalRankingMLOpsSQL

Similar roles

ML Engineering jobs
Orchard

Machine Learning Engineer

OrchardSan Francisco, CA +1

Build scalable ETL pipelines and ML infrastructure for processing massive farm image datasets from tractor-mounted cameras. Develop edge-deployed computer vision models for yield estimation, disease detection, and crop analysis using PyTorch and cloud tools, with 2+ years experience.

135k – 210k/yr
On-site2+ YOEML Engineering
Pinterest

Machine Learning Engineer II, Computer Vision Applied Science

PinterestSan Francisco, CA

Build and fine-tune vision-centric VLMs and generative models using Pinterest's visual-text datasets. Requires 2+ years industry computer vision experience and an M.S. or Ph.D.

139k – 286k/yr
Remote2+ YOEML Engineering
Pinterest

Machine Learning Engineer, Core Engineering

PinterestSan Francisco, CA +2

Build and improve machine learning models for Pinterest's recommendation systems across Homefeed, Ads, Search, and more. Requires 2+ years experience in ML methods like personalization and recommender systems, plus hands-on work with large-scale data pipelines.

139k – 286k/yr
Remote2+ YOEML Engineering
Traba

Software Engineer

TrabaNew York, NY

Build and own production AI agent systems (harnesses, evals, orchestration) on frontier LLMs for industrial supply chain workflows at Traba. Requires 1+ years shipping LLM/agent features to production, strong Python/TS skills, and high-agency in ambiguous customer environments.

140k – 200k/yr
On-site1+ YOEML Engineering
Notion

Software Engineer, Early Career

NotionSan Francisco, CA

Early-career engineer building and productionizing AI-powered features at Notion using LLMs and embeddings. Less than 2 years experience; strong fundamentals in algorithms, data structures, and distributed systems required.

130k – 150k/yr
HybridEntry levelML Engineering