Machine Learning Engineer, Early Stage Project

$160,000 - $210,000/Yr

Google - Mountain View, CA

Full-time - Mid Level

Mountain View, CA

Web Search Portals, Libraries, Archives, and Other Information Services

About the position

The Machine Learning Engineer role at X involves end-to-end development of innovative technologies utilizing large language models. The position focuses on optimizing prompts, implementing scalable data pipelines, and monitoring production performance. The engineer will collaborate with domain experts to refine technology and apply their expertise to create impactful solutions.

Responsibilities

• Explore, apply, and innovate state-of-the-art LLM & machine learning techniques to solve real-world problems
• Lead the design and implementation of ML data platforms and service frameworks, including labeling workflows and sampling strategies
• Design and implement robust, automated, production-level software using horizontally scalable components
• Work effectively with cross-functional teams of engineers, product managers, and domain experts
• Provide direction and focus in areas of high ambiguity

Requirements

• Bachelors/Masters in Computer Science, or equivalent practical experience
• Significant on-the-job experience developing and training ML models for product use-cases
• Proficiency in MLOps tasks such as deploying and versioning ML models in production environments and monitoring their performance
• Experience doing win/loss analysis on model results and systematically finding efficient ways to boost wins and reduce losses
• Experience with building data pipelines (e.g. Beam, Dataflow, Flume, Flink, Hadoop, Spark)
• Experience testing, maintaining or launching software products, including experience with software design and architecture
• Strong proficiency in coding with Python, Golang, or other equivalent languages
• Excellent written and verbal communication skills

Nice-to-haves

• Experience applying LLMs for large scale document processing, including tasks such as annotation, entity extraction, etc.
• Experience in ML modeling using TensorFlow, Pytorch or Jax/Flax
• Experience in Google Cloud Platform and its Vertex AI solution and capabilities
• Consistent track record of delivering high quality solutions to large, complex software problems
• Prior experience with setting the technical direction for small groups of engineers
• Experience working in start-up like environments

Benefits

• Base salary range of $160,000 - $210,000
• Bonuses
• Comprehensive benefits package