Senior Staff Machine Learning Engineer (AI Agent)

OmegaHires

1d • Remote

About The Position

As a Staff Machine Learning Engineer, your goal will be to take AI Agents from the realm of research and bring them into practical, real-world use cases. This includes developing and deploying proprietary LLMs, scaling AI solutions, and addressing key challenges such as evaluation and reliability. While we’re focused on real-world application rather than pure research, you’ll be working with some of the most advanced technologies in the GenAI space.

Requirements

• Bachelor’s Degree in Computer Science, Mathematics, or a related field; Master’s or Ph.D. preferred, or equivalent professional experience
• 7+ years of hands-on industry experience with AI and machine learning, preferably with 3+ years of experience working with LLMs in large-scale production environments
• Expert knowledge of machine learning concepts and methods, especially those related to NLP, Generative AI, and working with LLMs
• Proven leadership in designing and deploying AI solutions at scale, with a deep understanding of model optimization and real-time AI applications
• Extensive practical knowledge of modern machine learning frameworks and technologies (e.g., PyTorch, Tensorflow, Hugging Face, NumPy), as well as experience with distributed systems and cloud-based AI infrastructure
• Strong problem-solving and strategic thinking abilities, with a proven ability to lead cross-functional teams and work collaboratively to deliver innovative AI solutions in production
• A passion for driving AI adoption and pushing the boundaries of AI technology into real-world applications, with an ability to mentor junior engineers and influence strategic decisions across the organization

Responsibilities

• Focus on practical AI challenges such as improving reasoning, planning capabilities, and evaluation in real-world scenarios
• Collaborate with cross-functional teams including front-end and back-end software engineers to integrate AI Agents into Client customer solutions
• Lead initiatives to scale AI systems for production environments, ensuring performance and reliability across use cases
• Innovate and research ways to improve security, cost-efficiency, and reliability of AI systems

What This Job Offers

• Job Type: Full-time
• Career Level: Senior