ML Solutions Engineer

Summary

Join Arize AI, a leading AI observability and evaluation platform company, as an AI Solutions Engineer. You will work closely with sophisticated ML/GenAI teams, acting as a trusted advisor and building relationships with technical and business stakeholders. Responsibilities include advising on GenAI and ML best practices, giving product demos, running strategic business reviews, interfacing with pre-sales engineering, partnering with product and engineering teams, and spearheading new opportunities within existing accounts. The ideal candidate possesses experience as a Data Scientist, Machine Learning Engineer, or similar role, with expertise in public cloud environments, machine learning frameworks, and LLMs. Strong communication skills and a quick learning ability are essential. Arize offers a competitive salary, equity package, and comprehensive benefits, including medical, dental, vision, 401(k), unlimited PTO, generous parental leave, and more. The company is remote-first with optional in-person work in NYC or the Bay Area.

Requirements

• Previous experience working as a Data Scientist, Machine Learning Engineer, or as an Engineer working with ML models or GenAI applications in production
• Comfortable working in public Cloud environments (AWS, Azure, GCP)
• Knowledge of machine learning frameworks such as TensorFlow, PyTorch or Scikit-learn
• Knowledge of LLM / Agentic frameworks such as Llamaindex, LangGraph, and DSPy
• Understanding of ML/DS concepts, model evaluation strategies and lifecycle (feature generation, model training, model deployment, batch and real time scoring via REST APIs) and engineering considerations
• Understanding of GenAI concepts and application evaluation + development lifecycle
• Proficiency in a programming language (Python, JS/TS, Java, Go, etc)
• Strong Communication Skills - Ability to simplify complex, technical concepts
• A quick and self learner - undaunted by technical complexity of production ML deployments and welcome the challenge to learn about them and develop your own POV

Responsibilities

• Work closely with some of the most sophisticated ML / GenAI teams in the world
• You will act as a trusted advisor to our customers, while also building relationships with technical and business stakeholders
• Advise on GenAI and ML best practices
• Give ML and LLM product demos to technical and business stakeholders
• Run strategic business reviews for customers in partnership with our sales team
• Interface with our pre-sales engineering team to gather client goals and KPI’s
• Partner with our product and engineering teams to help drive the product roadmap
• Spearhead new opportunities within existing accounts to help drive expansions

Preferred Qualifications

• Previous engineering experience in: Data Science
• MLOps
• ML Frameworks
• LLM / Agentic frameworks
• Customer facing experience strongly preferred such as Solutions Architect, Implementation Specialist, Sales Engineer, Customer Success Engineer, Consultant, or Professional Service roles
• Prior experience working with applications deployed with Kubernetes
• Prior experience demoing technical products to both business and technical audiences

Benefits

• Medical
• Dental
• Vision
• 401(k) plan
• Unlimited paid time off
• Generous parental leave plan