Lead Production AI Engineer

Axial Search; United States; Vor Ort; USD 170.000 - 350.000; Vollzeit

Zusammenfassung

Axial Search is seeking an experienced senior ML engineer to design, build, and deploy production ML systems that address measurable business problems. You will own the full ML lifecycle—from data pipelines to production inference—working with product and data teams to translate needs into scalable solutions.

You will mentor junior engineers, contribute to infrastructure, and evaluate new tools that boost reliability and model quality.

Qualifikationen

• 5+ years of hands-on experience building and shipping ML systems in production environments.
• Fluency with at least one modern ML framework (PyTorch, TensorFlow, or JAX) and production ML infrastructure (model serving, experiment tracking, data pipelines).
• Ability to move from problem definition through model deployment—comfort with both research and engineering disciplines.
• Strong fundamentals in software engineering: version control, testing, debugging, and code review practices.
• Experience collaborating with cross-functional teams (product, data, operations) and communicating technical constraints and tradeoffs clearly.
• Self-directed learner with curiosity about new tools and techniques in AI/ML, balanced by pragmatism about what matters for the business.

Aufgaben

• Design, build, and deploy machine learning models and systems that solve measurable business problems at scale
• Own the full ML lifecycle: data pipeline design, model training and validation, optimization, and production inference
• Partner with product and data teams to define AI requirements, translate user needs into technical specifications, and integrate models into production applications
• Build and maintain ML infrastructure, monitoring, and testing practices that ensure model reliability and performance in production
• Troubleshoot model performance issues, conduct experiments to improve accuracy and latency, and document lessons learned
• Evaluate emerging AI tools, frameworks, and techniques; advocate for adoption where they measurably improve engineering velocity or model quality
• Mentor junior engineers and contribute to technical documentation and knowledge sharing across the engineering organization

Kenntnisse

Production ML experience

ML lifecycle ownership

Software engineering fundamentals

Cross-functional collaboration

Problem translation

Mentoring juniors

Tools

PyTorch

TensorFlow

JAX

Jobbeschreibung

Axial Search is seeking an experienced senior ML engineer to design, build, and deploy production ML systems that address measurable business problems. You will own the full ML lifecycle—from data pipelines to production inference—working with product and data teams to translate needs into scalable solutions.

You will mentor junior engineers, contribute to infrastructure, and evaluate new tools that boost reliability and model quality.