Senior/Staff Software Engineer, AI Insights

The role

You will design and build scalable agentic systems, backend services, and data-processing workflows. You will shape the architecture of APIs, databases, cloud infrastructure, and LLM-powered systems.

What you will do

• Architect and operate multi-step, tool-using agentic pipelines
• Build evaluation frameworks, golden datasets, and feedback loops
• Improve model performance through prompt and pipeline optimization, model selection, and fine-tuning
• Own the reliability of cloud-native services including deployment, observability, scalability, and on-call support
• Mentor engineers and lead cross-team technical initiatives

Required qualifications

• 5+ years building and operating production backend or data-intensive systems
• Production experience with LLM systems, including multi-step, tool-using, or agentic workflows
• Experience measuring and improving model quality, through evals, prompting, fine-tuning, or a mix
• Experience designing distributed systems, asynchronous workflows, and reliable data-processing services
• Working knowledge of data warehousing, ideally applied to versioned datasets, lineage, or feedback capture
• Experience running containerized applications on a major cloud platform ([AWS/GCP/Azure])

Nice to have

• Fine-tuning or distillation experience (SFT, LoRA, or provider tuning APIs)
• Experience with LLM observability and cost/latency optimization

Benefits

• Full Medical, Dental & Vision coverage for all single employees
• Axion covers an additional 70% of the excess for dependents
• HSA - Eligible Plan with monthly Axion contribution
• Unlimited PTO
• 401(k) Retirement Plan
• More Perks & Benefits included

About the company

Axion is a company founded in 2021 that provides an AI-powered platform to transform product quality and customer satisfaction in manufacturing. It is backed by investors including Bessemer Venture Partners and Salesforce Ventures.