AI Engineer (UK)

Writer is the full-stack generative AI platform delivering transformative ROI for the world’s leading enterprises. Our all-in-one solution makes it easy to deploy customized AI apps and workflows that accelerate growth, increase productivity, and ensure compliance. Writer’s suite of development tools is supported by Palmyra – Writer’s state-of-the-art family of LLMs – alongside our industry-leading graph-based RAG and customizable AI guardrails.

Writer is looking for an AI engineer with a strong software engineering background to join our expanding team of AI experts. We’re looking for a creative problem solver who has a deep understanding of NLP and ML technologies and who can help us create powerful and meaningful applications of AI. As an AI Engineer at Writer, you will play a pivotal role in developing and implementing state-of-the-art generative AI models and algorithms.

Your responsibilities:

• Collaborate closely with our broader engineering team, data scientists, and AI engineers to evolve our software architecture as we productize new AI-powered capabilities.
• Evaluate the performance of AI models & systems through rigorous testing and experimentation.
• Deploy LLM-based applications in production, evaluating and improving their performance over time.
• Collaborate with our skilled software engineers to seamlessly integrate AI-powered capabilities into production systems, ensuring scalability and efficiency.
• Stay up-to-date with the latest advancements in AI and machine learning research, and proactively suggest improvements to enhance our generative AI capabilities.
• Collaborate closely with cross-functional teams to understand business requirements and translate them into innovative AI solutions.
• Own the design, implementation, and maintenance of robust and scalable production retrieval and agentic systems.
• Implement strong testing and CI/CD practices that help us move with confidence in our AI system development.

Is this you?

• Bachelor’s degree in Computer Science, Engineering, Mathematics, related field, or equivalent experience.
• High level of coding proficiency using Python.
• 5+ years of professional experience in software engineering, AI/ML development including:

• Proficiency with production software (Python) and systems design.
• Machine learning algorithms and model development techniques.
• ML lifecycle tools like MLflow, dvc, weights & biases.
• Cloud deployment of ML systems.
• Professional experience with LLMs and large-scale models.
• Very strong software engineering skills with a track record of building scalable, distributed product machine learning systems.
• Strong analytical and problem-solving skills.
• Ability to communicate complex ideas and concepts effectively.
• Ability to work independently and collaboratively.

Preferred Skills:

• Experience building scalable applications with LLMs, using frameworks such as LangChain, LlamaIndex, Hugging Face, etc.
• Depth of knowledge with RAG implementation and improvements.

Benefits & perks:

• Generous PTO, plus company holidays.
• Medical, dental, and vision coverage for you and your family.
• Paid parental leave for all parents (12 weeks).
• Fertility and family planning support.
• Early-detection cancer testing through Galleri.
• Flexible spending account and dependent FSA options.
• Health savings account for eligible plans with company contribution.
• Annual work-life stipends for: Home office setup, cell phone, internet.
• Wellness stipend for gym, massage/chiropractor, personal training, etc.
• Learning and development stipend.
• Company-wide off-sites and team off-sites.
• Competitive compensation, company stock options and 401k.

Writer is an equal-opportunity employer and is committed to diversity.