Staff Machine Learning Engineer

San Francisco, California

  Machine Learning

Permanent

Our client are a leading innovator in the Financial Services sector and they are hiring a Staff Machine Learning Engineer to join the team in San Francisco. Utilizing your skills in Large Language Models (LLMs), the successful candidate will be involved throughout the entire process from model training to deployment, and will have a key role in expanding the company’s operations.

Responsibilities

  • As the Staff Machine Learning Engineer, you will develop and optimize both open-source and proprietary LLMs for tasks such as answering questions, summarization, reasoning and planning.

  • You will construct advanced Retrieval Augmented Generation (RAG) pipelines, encompassing rewriting, embedding fine-tuning, hybrid search, reranking and knowledge graphs.

  • Establish comprehensive evaluation frameworks and performance metrics for model assessment.

  • Deploy models in production environments, ensuring low latency, reliability and scalability.

  • Work closely with the product and software engineering teams to create end-to-end product systems.

Skillset

  • Ph.D. or Masters degree in in computer science, mathematics, statistics or related.

  • Extensive expertise in NLP/LLM with a deep understanding of the latest advancements in the field.

  • Hands-on proficiency with diverse LLM fine-tuning techniques (e.g., LORA), LLM inference frameworks (e.g., vLLM) and advanced RAG pipelines.

  • Strong command of LLM evaluation methods and performance metrics.

  • Two years of experience in developing applications with generative AI.

  • Proficient in machine learning/deep learning frameworks such as TensorFlow and/or PyTorch.

  • Skilled in programming languages including Python and SQL.

  • Thrives in a challenging, entrepreneurial, and fast-paced environment.

Salary: $150k – $250k DOE 

Interested? Apply now in the link below.

45833