Senior Machine Learning Engineer II
Axon · Seattle, WA · 1 wk ago
HybridEngineering$157k–$251k/yrFull-time
About the role
We are seeking a seasoned Machine Learning Engineer to join a new team building agentic video and multimodal reasoning systems. As a senior engineer on this team you will own the systems and infrastructure that turn cutting-edge research into safe, reliable, and scalable AI capabilities — supporting our company’s mission to accelerate justice, protect truth, and save lives.
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 — storage, processing, labeling, and retrieval.
- Build and scale training and evaluation infrastructure, and turn scientists’ evaluation methodology into automated, reproducible measurement systems.
- Build inference and serving systems for real-time and batch multimodal workloads, optimizing latency, throughput, and cost.
- Build the retrieval, indexing, and embedding infrastructure (vector search at scale) and the 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.
Requirements
- Bachelor’s Degree in Computer Science, Engineering, Physics, Mathematics or an equivalent highly technical field.
- 10+ 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.
Qualifications
- 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.
Skills
- Python programming.
- Experience with ML frameworks like PyTorch or TensorFlow.
- Experience with cloud infrastructure (AWS, GCP, or Azure).
- Experience with large-scale data systems handling video or other high-volume, high-dimensional multimodal data.
- Experience with distributed systems and container orchestration.
- Experience with video processing or video understanding pipelines.
- Experience with retrieval systems, vector search, and RAG infrastructure.
- Experience with serving foundation models, or building agentic systems and LLM tool-use orchestration.
- Experience 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.
- Employee Resource Groups (ERGs).
- Snacks in offices.
Pay
$156,750 - $250,800 USD
Schedule
Hybrid from Seattle, WA or London, UK