W

Writer

Enterprise AI platform for building agents

Staff/Senior Staff/Principal AI Research Scientist

AI ResearchPosted on Nov 12
Full-time

About the Job

Requirements

  • At least 7 years of hands-on machine learning research experience, including deep expertise in large language model pre-training and post-training, training models at scale, debugging distributed jobs, and shipping measurable improvements.
  • A Ph.D. in Computer Science, Machine Learning, NLP, or a related field, or equivalent demonstrated research experience with a strong portfolio of independent, published work.
  • Expert-level knowledge of post-training methods, including supervised fine-tuning, reinforcement learning from human feedback, reinforcement learning from AI feedback, direct preference optimization, group relative policy optimization, and related alignment and reasoning techniques, with a track record of applying them to production-grade systems.
  • Strong command of Python and PyTorch or JAX, with the engineering depth to build and scale training pipelines, evaluation infrastructure, and data synthesis workflows hands-on.
  • A meaningful publication record at competitive machine learning and artificial intelligence venues, such as NeurIPS, ICLR, ICML, ACL, or EMNLP, evidencing the ability to originate ideas and independently execute a multi-month research agenda.
  • Hands-on experience designing or evaluating agentic systems that plan, reason through multi-step tasks, use tools, and recover from errors, with an understanding of their limitations and ways to address them.
  • Ability to collaborate across research, engineering, and product, and communicate complex ideas clearly to technical and non-technical audiences.
  • Ability to challenge conventional thinking and pursue novel research directions.
  • Ability to drive projects end-to-end, take accountability for results, and focus on customer impact.

Responsibilities

  • Lead an independent, high-impact research agenda on large language models and agentic systems, owning projects from early hypothesis through model training, evaluation, and production deployment.
  • Design and execute large-scale post-training experiments using supervised fine-tuning, reinforcement learning from human feedback, reinforcement learning from AI feedback, direct preference optimization, and emerging alignment techniques to improve multi-step reasoning, planning, and tool use in enterprise agentic workflows.
  • Build novel evaluation benchmarks and methodologies that establish rigorous measures of model performance on complex, real-world enterprise tasks.
  • Develop scalable data synthesis and curation pipelines, including LLM-as-judge frameworks, synthetic data generation, and adversarial dataset construction.
  • Shape the model architecture and training roadmap by translating research insights into improvements to enterprise-grade large language models, in collaboration with research engineering and product teams.
  • Publish and present original research at leading venues, including NeurIPS, ICLR, ICML, and ACL, and represent the company in the scientific community.
  • Mentor and develop fellow researchers and engineers, helping set standards for scientific rigor, experimental design, and research culture.

About the Company

Writer

Enterprise AI platform for building agents

Writer provides an enterprise AI platform that lets organizations build, deploy, and supervise AI agents tailored to business processes. It runs on its Palmyra LLMs and uses a Knowledge Graph to connect internal data, grounding outputs in company context for accuracy. The platform includes guardrails for compliance and integrates with existing workflows via browser and app extensions. The goal is to help large organizations automate tasks and boost productivity while maintaining control and governance.

Company size
501-1,000
Company stage
Series C
Total funding
$326M
Headquarters
San Francisco, CA
Founded
2020

Recently Posted Jobs

Explore more opportunities at Writer

See more jobs