Skip to content
WriterWriterSan Francisco, CA

AI engineer

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.

152k – 316k/yr
Hybrid5+ YOEML Engineering

About the role

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

Skills

PythonPyTorchTensorFlowJAXMachine LearningDeep LearningNLPLLMsAWSGCPAzureMLOps

Similar roles

ML Engineering jobs
Brex

Software Engineer, Forward Deployed Agent Builder

BrexSeattle, WA

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating with systems and APIs, and measuring impact. Requires 4+ years experience shipping AI/automation with LLMs, agent frameworks, and databases.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Brex

Software Engineer, Forward Deployed Agent Builder

BrexSan Francisco, CA

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating systems, and measuring impact. Requires 4+ years experience with LLMs, agent frameworks, databases, and production AI systems.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Brex

Software Engineer, Forward Deployed Agent Builder

BrexNew York, NY

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating systems and APIs, and measuring impact. Requires 4+ years experience shipping AI/automation with LLMs, agent frameworks, and databases.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Notion

AI Applications Engineer

NotionSan Francisco, CA

Develop and prototype AI-driven solutions for GTM, Finance, and People teams, translating business problems into impactful prototypes using ML/AI and LLMs. Requires 4-8 years as Software Engineer or Data Scientist with production AI familiarity.

152k – 207k/yr
Hybrid4+ YOEML Engineering
Applied Intuition

Software Engineer - E2E Autonomy

Applied IntuitionSunnyvale, CA

Builds ML tools, infrastructure, and manages large datasets for end-to-end autonomy research and productionizing self-driving software. Works with AI research and engineering teams to scale GPU compute, data, and evaluation systems. Requires strong software generalist skills across ML stack.

153k – 222k/yr
On-siteML Engineering