Skip to content

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 – 200kMountain View, CAML EngineeringHybrid2+ YOE

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

Machine Learning Engineer

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 – 210kSan Francisco, CA +1ML EngineeringOn-site2+ YOESQLAWS

Machine Learning Engineer II, Computer Vision Applied Science

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 – 286kSan Francisco, CAML EngineeringRemote2+ YOELLMsRLHF

Machine Learning Engineer, Core Engineering

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 – 286kSan Francisco, CA +2ML EngineeringRemote2+ YOESparkHadoop

AI Engineer

DeepLearning.AI is seeking an AI Engineer to join their Learning Experience Lab. This role involves developing new AI products and features for their education platform, leveraging expertise in software development and machine learning, and building full-stack applications using GenAI tools.

130k – 175kMountain View, CAML EngineeringOn-siteEntry levelRAGCSS

AI/ML Engineer

Builds and deploys production AI systems including agentic workflows, RAG pipelines, and LLM integrations for search, personalization, fraud detection, and generative features at a marketplace platform. Requires 2+ years AI/ML production experience, Python proficiency, and agentic framework expertise.

140k – 160kBellevue, WAML EngineeringOn-site2+ YOERAGSQL