# AI engineer

**Company:** [Writer](https://hotfix.jobs/companies/writer)
**Location:** San Francisco, CA, New York, NY
**Role:** ML Engineering
**Salary:** $152k – $316k/yr
**Experience:** 5+ years
**Skills:** Python, PyTorch, TensorFlow, JAX, Machine Learning, Deep Learning, NLP, LLMs, AWS, GCP, Azure, MLOps
**Posted:** 2026-05-18

> Build and deploy scalable AI applications and intelligent agents for enterprise customers. Requires 5+ years of experience with Python, ML frameworks (PyTorch/TensorFlow/JAX), LLMs, and cloud platforms.

## Job Description

## What you'll do
- Architect, develop, and deploy high-performance, scalable AI applications into production environments, ensuring robust integrations with our end-to-end platform.
- Drive the development of intelligent agents and AI-powered features, translating complex research into practical, impactful solutions for our customers.
- Collaborate closely with research scientists, data scientists, and product managers to define technical requirements and deliver features that address critical business needs.
- Contribute to the research and evaluation of emerging AI technologies, frameworks, and tools to maintain WRITER's leadership in the enterprise generative AI space.
- Champion responsible AI practices, including bias detection, explainability, and safety alignment, making sure our AI is trustworthy and ethical.

## What you need
- 5+ years of professional experience building and deploying machine learning or AI systems in a production environment.
- Proficiency in Python and experience with relevant AI/ML frameworks such as PyTorch, TensorFlow, or JAX.
- A strong foundation in machine learning principles, deep learning architectures, and natural language processing, particularly with large language models.
- Experience with cloud platforms (e.g., AWS, GCP, Azure) and MLOps tools for managing the AI lifecycle.
- Demonstrated ability to write clean, efficient, and well-tested code for scalable software systems.
- Exceptional problem-solving skills and a passion for tackling complex technical challenges with creative AI solutions.

## 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 (16 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 wellness, learning and development
- Company-wide off-sites and team off-sites
- Competitive compensation, company stock options and 401k

## Similar roles

- [Software Engineer, Forward Deployed Agent Builder](https://hotfix.jobs/jobs/715a9748-6a88-4317-8629-58099d697b24) - Brex - Seattle, WA - $152k – $240k/yr
- [Software Engineer, Forward Deployed Agent Builder](https://hotfix.jobs/jobs/8ccee115-678f-40d5-9064-b03d724f52a4) - Brex - San Francisco, CA - $152k – $240k/yr
- [Software Engineer, Forward Deployed Agent Builder](https://hotfix.jobs/jobs/ae0bb066-9d72-4c0e-b443-4a6366ab595d) - Brex - New York, NY - $152k – $240k/yr
- [AI Applications Engineer](https://hotfix.jobs/jobs/a2a15c15-1362-470f-976c-916d0964e3b1) - Notion - San Francisco, CA - $152k – $207k/yr
- [Software Engineer - E2E Autonomy](https://hotfix.jobs/jobs/e3b6ebe4-5501-421b-ae1a-a736896881af) - Applied Intuition - Sunnyvale, CA - $153k – $222k/yr

**Apply:** https://hotfix.jobs/jobs/cbe50e1e-cfc8-40e3-82dc-cb84345430d6
**Canonical:** https://hotfix.jobs/jobs/cbe50e1e-cfc8-40e3-82dc-cb84345430d6