RemoteLeaf Logo
Log in Sign up

AI Tech Lead - Staff Machine Learning Engineer

55 minutes ago
United States
Full-time
Lead
AI Engineer
Software Development
Apache Airflow AWS Azure Docker GCP Kubernetes LLM Machine Learning MLflow Python PyTorch System Design Vertex AI
Apply Now
Sumo Logic

Sumo Logic

Sumo Logic offers top-tier cloud monitoring, log management, and Cloud SIEM tools for web and SaaS apps, empowering businesses with real-time insights and high-quality software delivery.

Internet Software & Services
251-1K
Founded 2010
View All Jobs 7

Description

  • Lead technical evaluation and adoption of agentic AI platforms and frameworks, including Anthropic, LangChain/LangGraph, AWS Bedrock, and emerging tools.
  • Architect, prototype, and productionize multi-agent AI systems for detection, triage, investigation, and response workflows.
  • Own core agent architecture components such as planning, execution, tool orchestration, memory, context engineering, and long-running workflows.
  • Lead offline and online evaluation pipelines for AI agents, including golden datasets, synthetic data generation, and human/LLM-based judging.
  • Drive LLM fine-tuning and alignment efforts to improve domain-specific reasoning, accuracy, and reliability.
  • Design scalable LLMOps and AI infrastructure for inference routing, latency optimization, cost control, and production observability.
  • Partner with product, security, and data platform teams to deliver end-to-end AI capabilities from prototype to customer-facing systems.
  • Provide technical direction and mentorship to AI engineers working on agentic AI and LLM systems.
  • Define and implement best practices for AI safety, reliability, evaluation, and monitoring in production.
  • Set technical direction for ambiguous problems and drive delivery across teams.

Requirements

  • B.Tech, M.Tech, or Ph.D. in Computer Science, Machine Learning, Data Science, or a related technical field.
  • 5+ years of hands-on industry experience building, operating, and leading production ML/AI systems.
  • Strong foundation in machine learning, distributed systems, data pipelines, and large-scale system design.
  • Deep understanding of LLMs, prompt engineering, context engineering, agentic AI design patterns, and reasoning workflows.
  • Strong proficiency in Python and modern ML/AI ecosystems.
  • Experience designing and operating evaluation frameworks for ML/LLM systems, including offline and online evaluation.
  • Proven ability to lead complex technical initiatives across teams and influence architecture decisions.
  • Excellent communication skills and ability to translate complex AI systems into business impact.
  • Hands-on experience building and scaling agentic AI systems or multi-agent architectures in production (preferred).
  • Experience with modern agent frameworks such as LangGraph, LangChain, CrewAI, or similar (preferred).
  • Experience with foundation model platforms such as Anthropic, OpenAI, AWS Bedrock, or Vertex AI (preferred).
  • Experience with LLM fine-tuning pipelines such as SFT, RLHF/RLAIF, preference learning, or domain adaptation (preferred).
  • Strong background in LLMOps, including inference optimization, latency/cost management, observability, and production monitoring (preferred).
  • Experience with tools such as PyTorch, MLflow, Airflow, Docker, Kubernetes, and cloud platforms like AWS, GCP, or Azure (preferred).
  • Experience applying AI/ML to security, observability, or large-scale log/telemetry data is a strong plus.
  • Must be authorized to work in the United States at the time of hire and for the duration of employment.
  • No non-immigrant visa sponsorship is available for this position.

Benefits

  • Eligible roles may participate in bonus or commission plans.
  • Eligible roles may receive equity awards.
  • Benefits offerings are available as part of the compensation package.
  • Compensation is variable based on role level, skills, qualifications, location, and experience.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Klaviyo

Senior AI Engineer, ARIA Team

Klaviyo 1K-5K IT Services

Klaviyo’s ARIA team is hiring a Senior AI Engineer to build and scale production AI systems that power internal automation, knowledge retrieval, and employee workflows.

United States Full-time Senior AI Engineer
$112k-$168k
AWS Azure CI/CD CircleCI Django Docker FastAPI Flask GCP GitHub Actions JavaScript Jenkins Kubernetes LLM Machine Learning Pulumi Python Terraform TypeScript
10 minutes ago
NEORIS

AI Engineer pharma sector

NEORIS 5K-10K Internet Software & Services

NEORIS, now part of EPAM, is seeking an AI Engineer for its Connected Insights team to develop production-grade AI systems that generate business and scientific insights for strategy-focused work.

Spain Full-time Senior AI Engineer
Agile CloudFormation Git GPT Kubeflow LLM Machine Learning Microservices MLflow Python Serverless Terraform
10 minutes ago
JustMarkets

Head of Data

JustMarkets 1-10 Capital Markets

Head of Data at a fintech company to unify data engineering, data science, quant, and analytics into one strategic function that delivers scalable data platforms and AI-driven business solutions.

Anywhere Full-time Lead AI Engineer Data Scientist
Apache Spark CI/CD ClickHouse Feature Engineering LLM Machine Learning MLflow MLOps Python Scikit-learn SQL XGBoost
10 minutes ago
Monzo

Machine Learning Manager, Borrowing

Monzo 1K-5K Banks

Monzo is hiring a Borrowing Machine Learning Scientist to lead a small team in improving credit decisioning and model outcomes across its borrowing products.

United Kingdom Full-time Lead Machine Learning Engineer Program Manager
$127k-$203k
AWS dbt GCP Go Machine Learning Microservices Python SQL Statistics
10 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers