Senior Machine Learning Engineer I
Axon
Opportunity published from Greenhouse · applications stay with the original hiring source.
About the opportunity
At Axon, we are on a mission to Protect Life, pursuing society's most critical safety and justice issues with our ecosystem of devices and cloud software. Life at Axon is fast-paced, challenging, and meaningful, giving you the opportunity to take ownership and drive real change.
We are seeking a seasoned Machine Learning Engineer to join a new team building agentic video and multimodal reasoning systems.
Working closely with research scientists, you will own the systems and infrastructure that turn cutting-edge research into safe, reliable, and scalable AI capabilities supporting our mission to accelerate justice, protect truth, and save lives.
Key responsibilities
- Own end to end the systems and infrastructure that take multimodal and video models from research prototype to reliable, scalable production.
- Design and operate data and ingest pipelines for large-scale video and multimodal corpora including storage, processing, labeling, and retrieval.
- Build and scale training and evaluation infrastructure, turning scientists' evaluation methodology into automated, reproducible measurement systems.
- Build inference and serving systems for real-time and batch multimodal workloads while optimizing latency, throughput, and cost.
- Build retrieval, indexing, and embedding infrastructure (vector search at scale) and orchestration and tool-use plumbing behind agentic reasoning over video and multimodal data.
- Ensure the platform is secure, privacy-preserving, and responsible by design.
- Set technical direction for the team's systems, raise the engineering bar, and mentor other engineers.
Qualifications
- Bachelor's Degree in Computer Science, Engineering, Physics, Mathematics, or an equivalent highly technical field.
- 7+ years of software engineering experience and a proven track record of architecting, operating, and maintaining large-scale distributed platforms in production.
- Deep experience building systems across the ML lifecycle: data pipelines, training and evaluation infrastructure, model serving and inference at scale, and production monitoring.
- Experience with large-scale data systems handling video or other high-volume, high-dimensional multimodal data.
- Strong software engineering fundamentals and proficiency in Python, along with familiarity with ML frameworks such as PyTorch or TensorFlow.
- Hands-on experience operating cloud infrastructure at scale (AWS, GCP, or Azure), including distributed systems and container orchestration.
- Excellent problem-solving skills and the ability to dive into system architecture, design, performance metrics, code, test plans, project plans, deployments, and operations.
- Strong communication skills and comfort partnering closely with scientists, engineers, and product managers.
Nice to have
- Master's or PhD in Computer Science, Engineering, or an equivalent highly technical field.
- Experience building video processing or video understanding pipelines at scale.
- Experience with retrieval systems, vector search, and RAG infrastructure.
- Experience serving foundation models, or building agentic systems and LLM tool-use orchestration.
- Familiarity with responsible AI, de-identification, and privacy-preserving techniques.
Benefits
- Competitive salary and 401k with employer match
- Discretionary paid time off
- Paid parental leave for all
- Medical, Dental, Vision plans
- Fitness Programs
- Emotional & Mental Wellness support
- Learning & Development programs
Job details
- Location: Seattle, WA
- Remote/hybrid status: Hybrid schedule based out of the Seattle, WA office, working onsite Tuesdays through Fridays with the flexibility to work remotely on Mondays, unless there is an approved workplace accommodation.
Base Pay Range
$150,750 — $241,200 USD (actual base pay depends on level, function, training, transferable skills, work experience, business needs, and geographic market). Total compensation also includes bonus and stock awards.
- Reports to: Tom Faulhaber, Director of Software Engineering, AI
About the company
Axon is a total compensation company creating an ecosystem of devices and cloud software to protect life, capture truth, and accelerate justice. We think big with a long-term view to reinvent the world to be a safer, better place.
Additional information
Some roles may require legal eligibility to work in a firearms environment. All Axon employees must be aware of and committed to appropriate environmental, health, and safety regulations, policies, and procedures.
Application: UpVagas lists this opportunity from the hiring company or its recruitment platform. The application is completed at the original source.
Full listing
UpVagas condensed repetitive or legal boilerplate to make this page easier to read. The original hiring source contains the complete job description.
