Staff+ AI/ML Engineer

  • San Mateo, CA
  • Posted 3 days ago | Updated 3 days ago

Overview

On Site
Depends on Experience
Full Time

Skills

PyTorch
TensorFlow
Transformer Models
Large Language Models (LLMs)
Model Training
Fine-Tuning
Distillation
Supervised Fine-Tuning (SFT)
Reinforcement Learning (RL)
Policy Optimization
Retrieval
Search
Reasoning
Tool Calling
GitHub
Documentation Systems
CI/CD Tools
Cursor
Claude Code
Coderabbit
Warp
Startup Environment
Autonomous Work

Job Details

Job Description:
We are seeking a highly skilled Staff+ Engineer with deep expertise in model training and large language models (LLMs) to join our early-stage team. You will work directly with the founders to design, train, and scale cutting-edge AI systems and play a critical role in shaping product direction.

Responsibilities:

  • Explore and apply the latest research in retrieval, reasoning, search, and tool calling to build advanced AI agents

  • Define and run reinforcement learning (RL) experiments using data from tools such as GitHub, documentation, and CI/CD systems

  • Collaborate with founders and early customers to validate feasibility and guide product strategy

Requirements:

  • Master s/PhD in Computer Science, Machine Learning, Data Science, or related field

  • 8+ years of hands-on experience in research/engineering with PyTorch or TensorFlow and transformer-based models

  • Strong knowledge of LLM training/fine-tuning methods (distillation, SFT, RL, policy optimization)

  • Ability to work autonomously in a fast-paced startup environment

  • Familiarity with tools like Cursor, Claude Code, Coderabbit, Warp, etc.

Work Arrangement: Onsite San Mateo, CA

If you are passionate about building next-generation AI products and thrive in fast-moving environments, we d love to connect with you!

Employers have access to artificial intelligence language tools (“AI”) that help generate and enhance job descriptions and AI may have been used to create this description. The position description has been reviewed for accuracy and Dice believes it to correctly reflect the job opportunity.