Staff Machine Learning Engineer, Retrieval

The goal of a Staff Machine Learning Engineer at Scale is to lead the design and deployment of agentic AI systems that operate in real-world, mission-critical government environments. On the Public Sector team, you’ll work at the intersection of agentic ML, systems engineering, and applied research, building foundational infrastructure that enables AI systems to reason, plan, and act reliably at national scale. Our Public Sector ML Team partners directly with U.S. defense and intelligence agencies to deploy AI into classified and regulated environments. Through flagship programs like Donovan and Thunderforge , we are advancing the next generation of agentic AI for geospatial reasoning, planning, and decision support. Staff Machine Learning Engineers play a central role in setting technical direction, owning core architectures, and translating ambitious ideas into production systems trusted by government operators. You will: Lead the architecture and implementation of agentic AI systems, with a focus on long-horizon reasoning, orchestration, and system-level reliability. Build and scale agents that perform complex geospatial reasoning, including interpreting, generating, and reasoning over maps and spatial data. Design and improve retrieval systems across large collections of static and semi-structured documents, enabling agents to surface high-signal context efficiently. Fine-tune and evaluate embedding models to improve recall and precision for mission-critical datasets. Design memory systems that allow agents to persist state, operate over long contexts, and learn from prior interactions. Own and evolve shared agentic infrastructure and core libraries, enabling reuse across teams, products, and Public Sector contracts. Define evaluation strategies for agentic systems, including robustness testing, failure-mode analysis, and regression testing in production environments. Partner closely with engineering managers, product leaders, and researchers to scope high-impac

Join Adobe Security Engineering and help build the ML and generative AI capabilities that protect Adobe, our products, and our customers. Our team sits where cybersecurity, large-scale data, and AI meet, building models for anomaly detection, threat detection, investigation, and security analytics across some of Adobe's largest datasets. As a Staff Machine Learning Engineer, you'll design, build, and scale production ML systems spanning deep learning, behavioral modeling, embeddings, and agentic AI. You'll write code, train models, run experiments, and help shape the architecture the broader team builds on. You'll also partner closely with data, platform, and security engineers across Adobe. The Challenge Architect end-to-end ML systems for high-volume security data, turning experiments into reusable capabilities. Own the ML lifecycle, from feature engineering through training, deployment, monitoring, and retraining. Build LLM and agentic AI capabilities for security investigation, including retrieval-augmented generation. Design evaluation frameworks that measure model quality and help analysts trust and validate results. Work with data and platform engineers to scale pipelines and resolve system bottlenecks together. Mentor engineers and help guide technical direction through design reviews and hands-on collaboration. What you bring Experience running production ML systems from early experimentation through sustained operation at scale. A strong background training models with PyTorch and transformers, including behavioral modeling and anomaly detection. Comfort with distributed compute like Spark, cloud platforms like AWS, and MLOps tools like MLflow. Experience building with LLMs or generative AI, along with evaluating how they behave in production. Strong Python and SQL skills, and solid software engineering habit

About Snorkel Snorkel AI is the frontier AI data lab, helping teams build the data and environments behind high-performing frontier and agentic AI. We combine technology with research-driven AI data development to create datasets, benchmarks, evals, and custom solutions for real-world AI systems. Founded out of the Stanford AI Lab in 2019, Snorkel works with leading AI labs and enterprises to move from better data to better outcomes. Excited to help us redefine how AI is built? Apply to be the newest Snorkeler! The Role As an Applied AI Engineer, you’ll research and utilize state-of-the-art Gen AI and machine learning (ML) techniques to successfully deliver solutions to our customers. You will work directly with our customers to understand their business and technical needs and design and deliver AI solutions to solve them. You will also help define Snorkel’s Applied AI tooling by translating repeatable real-world challenges into reusable solution recipes, workflows, best practices, and platform-level capabilities that become part of Snorkel’s next generation of AI tooling. We move fast and are constantly prototyping and innovating new ways to deliver value to our customers. This position is ideal for someone who enjoys solving complex problems, bridging the gap between AI technology and business value, working directly with customers, keeping up-to date with AI research, and standardizing bespoke solutions into internal recipes and staying naturally curious about the infrastructure that underpin the Applied AI stack end-to-end. Main Responsibilities Partner with customers to build and deploy impactful Gen AI and machine learning solutions, from use case scoping and data exploration to model development and deployment. This will involve designing custom approaches using state-of-the-art tools, with the goal of delivering real business value and informing the evolution of Snorkel’s tooling. Develop and implement state of the art AI systems such as retrieval-aug

About Scale AI Scale AI is the data foundation for AI, helping organizations build and deploy reliable production AI applications. We partner with the world's leading enterprises and government organizations to accelerate their AI transformation through frontier AI systems that solve real business problems. Every day, we work with organizations across finance, healthcare, manufacturing, media and telecommunications to build production AI agents that automate complex workflows, help humans, reason over enterprise knowledge, and operate safely at scale. The Opportunity Applied AI is moving faster than ever. New foundation models, reasoning techniques, agent architectures, and research papers emerge every week. Yet building AI systems that reliably solve real-world problems remains one of the hardest engineering challenges. As a Staff Frontier Agent Engineer (Applied AI) , you'll bridge the gap between cutting-edge AI research and production deployment. You'll work directly with enterprise customers to design, evaluate, and deploy intelligent systems that combine frontier models with structured knowledge, retrieval, traditional machine learning, and enterprise software. Unlike traditional ML roles that focus on a single model or product, you'll work across a diverse portfolio of AI challenges spanning multiple industries and use cases. You may build a multi-agent research system and then participate in designing a customer intelligence platform, a healthcare copilot, or an autonomous workflow for a Fortune 100 company. If you enjoy reading new AI papers, experimenting with the latest models, and shipping production systems that create measurable business impact, you'll fit right in. What You'll Build Frontier AI Systems Design and deploy production AI agents that leverage the latest advances in large language models, reasoning, retrieval, memory, and tool use. Architect intelligent systems that combine LLMs, traditional machine learning, structured knowledge, ent

We’re building a world of health around every individual — shaping a more connected, convenient and compassionate health experience. At CVS Health®, you’ll be surrounded by passionate colleagues who care deeply, innovate with purpose, hold ourselves accountable and prioritize safety and quality in everything we do. Join us and be part of something bigger – helping to simplify health care one person, one family and one community at a time. POSITION SUMMARY CVS Health is looking for hands-on, passionate people who want to join a high energy and growing team to make a difference in customers’ lives and who want to be on the forefront of digital innovation that aims to reinvent what a pharmacy and a health care company can be in the digital world. Currently, we are seeking a Staff Software Engineer – Search / AI who as a Senior technical leader, be responsible for driving architecture, design, and delivery of scalable, cloud-native platforms built on microservices architecture and AI capabilities. This role combines deep hands-on engineering with strategic leadership to build intelligent, distributed systems. The right candidate will be a strong analytical thinker and be able to simplify complex problems, processes or projects into component parts explore and evaluate them systematically. We love to collaborate and help each other and we want someone to share that ideology. Expectations for the Role Drive enterprise architecture and technical strategy with strong focus on microservices-based design and AI platform engineering Design and develop highly scalable microservices architectures, including APIs, domain-driven services, and event-driven systems Lead the development and integration of AI/ML solutions, including LLMs, Retrieval-Augmented Generation (RAG), and agentic frameworks Develop sc

At ClickUp, we're building the future of work: the first truly converged AI workspace unifying tasks, docs, chat, calendar, and enterprise search, all supercharged by context-driven AI. We are an AI-native company. Every team member is expected to leverage AI daily, and we evaluate AI fluency as part of our hiring process. Join us and help redefine what's possible. 🚀 ClickUp is seeking a talented and experienced Senior Search Engineer to join our team and help revolutionize our search capabilities. As a key member of our engineering team, you'll be responsible for optimizing and enhancing our search functionality, which is a critical feature of our platform and underpins our AI efforts. Your impact at ClickUp As a Senior Engineer on the Search squad, you'll be responsible for designing, optimizing, and scaling our search infrastructure. Our platform is built as backend services on top of Postgres and OpenSearch, focused on real-time search ingestion and serving. You'll work hands-on with these systems to ensure users can instantly find exactly what they need across their workspace. Your expertise will directly contribute to making ClickUp the most intuitive and responsive productivity platform available. Core Responsibilities Design and implement robust search solutions that scale with our rapidly growing user base Improve search relevance, accuracy, and speed to deliver the most relevant results to users at blazing fast speeds Improve our real-time indexing pipelines to ensure search results remain up-to-date Create measurement frameworks to evaluate and improve search quality Build and enhance vector search capabilities to power next-generation search experiences Collaborate with AI, backend, and product teams to integrate search into new features Troubleshoot complex search-related issues at scale Design and implement robust search solutions that scale with our rapidly growing user base Improve search relevance, accuracy, and speed to deliver the most relevant r