Staff Machine Learning Engineer

Staff Machine Learning Engineer | Enterprise AI Platform | Remote (U.S.)

The Company

Our client is an innovative AI platform company developing next-generation enterprise artificial intelligence solutions that enable organizations to securely deploy AI using their own proprietary data. Their mission is to build efficient, scalable, and private AI systems that deliver meaningful business outcomes while challenging conventional approaches to machine learning.

This is an opportunity to join a rapidly growing engineering team where you'll help shape the future of enterprise AI, working on cutting-edge machine learning technologies alongside highly experienced researchers and engineers.

• Join a fast-growing AI company at the forefront of enterprise Generative AI
• Work on advanced LLM and machine learning technologies that solve real-world problems
• High-impact role with genuine technical ownership and leadership opportunities
• Collaborative, innovation-driven engineering culture with exceptional career growth
• Competitive salary, equity package, comprehensive healthcare, flexible time off, and ongoing learning support

The Role

Our client is seeking an experienced Staff Machine Learning Engineer to lead the development of advanced AI systems powered by Large Language Models (LLMs). You'll play a key leadership role across product development, research, and engineering, driving the delivery of production-ready AI solutions while mentoring other engineers and helping shape the technical direction of the team.

This position offers the opportunity to work across the full machine learning lifecycle, from research and optimization through to deployment and platform integration.

• Lead the design, development, and optimization of Large Language Model (LLM) applications and AI-powered products
• Drive key initiatives across Retrieval-Augmented Generation (RAG), AI agents, tool calling, model quantization, and on-device inference optimization
• Partner with software engineering, product, and platform teams to deliver scalable AI capabilities into production
• Research and implement state-of-the-art machine learning techniques to improve model performance, efficiency, and reliability
• Mentor engineers, provide technical leadership, and establish engineering best practices across the AI organization

Your Experience

We're looking for an experienced Machine Learning Engineer with a strong research and engineering background in Generative AI. You'll combine deep technical expertise with proven technical leadership, enjoying the opportunity to mentor engineers while helping shape the direction of high-performing AI teams.

• 6+ years of experience in Machine Learning, Artificial Intelligence, or Deep Learning engineering
• Strong commercial experience developing and deploying Large Language Models (LLMs) and Generative AI applications
• Expert-level Python programming skills with hands-on experience using PyTorch and/or TensorFlow
• Experience building production AI systems across areas such as RAG, model optimization, inference, agentic AI, or tool-calling frameworks
• Demonstrated experience leading technical teams, mentoring engineers, and driving engineering best practices across complex AI projects
• Strong stakeholder management and communication skills, with the ability to lead cross-functional technical initiatives

Desirable experience:

• Experience hiring, building, or growing high-performing Machine Learning or AI engineering teams
• Advanced degree (Master's or Ph.D.) in Computer Science, Machine Learning, Artificial Intelligence, or a related discipline
• Experience with cloud platforms including AWS, Azure, or Google Cloud Platform (GCP)
• Knowledge of distributed data technologies such as Spark or Hadoop
• Experience with Docker, Kubernetes, and modern MLOps practices
• Published research, conference presentations, or notable contributions within the AI or Machine Learning community