AI engineer

About this role

Writer is looking for an AI engineer with a strong software engineering background to join our expanding team of AI experts. 

At Writer, we believe in using the power of AI to unlock the potential of the enterprise. With the help of our AI engineers, we can continue to build the most advanced language model available in the industry and revolutionize how companies interact with AI. 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'll play a pivotal role in developing and implementing state-of-the-art generative AI models and algorithms. Collaborating closely with our diverse and dynamic team of software engineers, data scientists, and researchers, you'll be able to design and deploy AI solutions that drive our innovative products. 

If you’re passionate about using AI to transform the enterprise, then we want to hear from you.

‍♀️ 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

  #LI-Remote

Benefits & perks (US Full-time employees)

• 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. We don't make hiring or employment decisions based on race, color, religion, creed, gender, national origin, age, disability, veteran status, marital status, pregnancy, sex, gender expression or identity, sexual orientation, citizenship, or any other basis protected by applicable local, state or federal law. Under the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

By submitting your application on the application page, you acknowledge and agree to Writer's Global Candidate Privacy Notice.