Lead AI Research Scientist — LLMs & Agent Systems

present of AI; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Writer is seeking a researcher who has trained large language models at scale and can own a project from concept through training, evaluation and release. The ideal candidate writes PyTorch or JAX code and has published at top ML venues.

They should have real experience with agents that plan, use tools, and recover from mistakes in production settings. The role focuses on advancing how models reason, plan and act across multi-step tasks within enterprise data environments, with strong emphasis

Qualifications

• Experience training large language models at scale.
• Strong knowledge of post-training methods.
• Experience with enterprise agents that plan, use tools, and recover from mistakes.
• Publications in top ML venues preferred.

Responsibilities

• Own a project from idea to training, testing and release.
• Experiment with agents that plan, use tools and recover from mistakes.
• Publish findings in top ML venues.

Skills

LLM Training

Post-training methods

Agent planning & tools

Tools

PyTorch

JAX

Job description

Writer is seeking a researcher who has trained large language models at scale and can own a project from concept through training, evaluation and release. The ideal candidate writes PyTorch or JAX code and has published at top ML venues.

They should have real experience with agents that plan, use tools, and recover from mistakes in production settings. The role focuses on advancing how models reason, plan and act across multi-step tasks within enterprise data environments, with strong emphasis