Lead Software Engineer Machine Learning Jobs

At Lyft, our mission is to improve people's lives with the world's best transportation. To accomplish this, we start with our community by creating an open, inclusive, and diverse organization. About the Team The Risk Tech engineering organization is committed to tangibly reducing accident frequency, saving lives, and managing costs to enhance the safety and affordability of rides. Claim Management is a core financial function for Lyft. Each claim touches complex workflows, multiple stakeholders, sensitive data, financial reserves, regulatory processes, and significant financial liabilities. This role offers the opportunity to define a leading claims management system for the industry. Our vision is to establish a single, Unified Risk Platform where comprehensive claims workflows across all business lines are efficiently administered, communications are consolidated, and data is structured to facilitate data-driven insights and decisions.This enables cost-efficient claims operations, mitigates risks and expenses as Lyft scales, and ensures people receive assistance proactively and accurately. About the Role We are seeking a Senior Software Engineer to contribute to the technical direction, drive architectural decisions, and lead the development of a highly reliable, scalable, and intelligent Risk Management Information System that powers our insurance platform. You will collaborate with passionate colleagues from Engineering, Data Science, Product and Claim Operations to deliver end-to-end solutions. Responsibilities: Define and drive the long-term technical roadmap for claims management systems, aligning priori

At Lyft, our mission is to improve people's lives with the world's best transportation. To accomplish this, we start with our community by creating an open, inclusive, and diverse organization. About the Team The Risk Tech engineering organization is committed to tangibly reducing accident frequency, saving lives, and managing costs to enhance the safety and affordability of rides. Claim Management is a core financial function for Lyft. Each claim touches complex workflows, multiple stakeholders, sensitive data, financial reserves, regulatory processes, and significant financial liabilities. This role offers the opportunity to define a leading claims management system for the industry. Our vision is to establish a single, Unified Risk Platform where comprehensive claims workflows across all business lines are efficiently administered, communications are consolidated, and data is structured to facilitate data-driven insights and decisions.This enables cost-efficient claims operations, mitigates risks and expenses as Lyft scales, and ensures people receive assistance proactively and accurately. About the Role We are seeking a Senior Software Engineer to contribute to the technical direction, drive architectural decisions, and lead the development of a highly reliable, scalable, and intelligent Risk Management Information System that powers our insurance platform. You will collaborate with passionate colleagues from Engineering, Data Science, Product and Claim Operations to deliver end-to-end solutions. Responsibilities: Define and drive the long-term technical roadmap for claims management systems, aligning priori

ABOUT BASETEN Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F , led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products. THE ROLE As a Software Engineer at on the Training Infrastructure team, you'll architect and lead development of our training platform, supporting top tier research engineers and model developers. You'll make key technical decisions for the infrastructure enabling developers to deploy, scale, and monitor their workloads with high performance and reliability. You’ll own scheduling, storage, networking, reliability, and observability of technical systems in the training stack EXAMPLE INITIATIVES Take a look at what we’ve built so far: Overview of the product so far Training docs overview Story of the Training product Research we've done RESPONSIBILITIES Design and architect scalable infrastructure systems for our ML training platform (e.g. scheduling, storage, and networking) Partner closely with developers and research engineers to translate complex training requirements into technical solutions Design and architect a global training scheduler Design and architect reinforcement learning systems and continuous learning pipelines Drive long-term improvements to improve reliability of systems and velocity of development Partner closely with SRE and Capacity teams to unlock state of the art training infrastructure Make critical architectural decisions balancing performance with system reliability Lead technical discussions and mentor junior engineers on infrastructure best practices Contribute to long-term technical strateg

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. We’re redefining Privileged Access Management (PAM) from the ground up, purpose-built for Cloud, SaaS, Databases, Containers, and any virtualized environment. Our mission is to simplify and secure workforce access with seamless, secure-by-default workflows that adapt dynamically to modern infrastructure. We eliminate standing privileges, enforce least privilege, and embed Zero Trust principles into every access workflow by default. About the Role We are seeking a Principal Backend Engineer (P5) to serve as the technical leader and compass for our newly established engineering pod in India. Operating at the intersection of identity, networking, and security infrastructure , you will be responsible for tackling highly complex, vaguely specified problems without day-to-day oversight. In this role, you will champion the technical execution of your team. You will lead the design and implementation of secure database and network device connectors (routers, switches, firewalls) on top of our core Zero Standing Privileges (ZSP) platform. You will work closely with our local Technical Team Lead to mentor mid-level engineers, while collaborating closely with global Tech Leads to ensure architectural alignment. What You’ll Be Doing Lead Technical Strategy & Execution: Turn high-level, complex PAM and network access problems into clear, modular technical designs that your team can execute. Own Connector Ecosystems: Design, architect, and optimize resilient connecto

At Lyft, our purpose is to serve and connect. We aim to achieve this by cultivating a work environment where all team members belong and have the opportunity to thrive. The Global Support & Partnerships team is a critical and unique part of helping Lyft succeed in our purpose to Serve and Connect. We support Lyft's international growth by developing integration platforms that seamlessly link global customers to an exceptional customer care experience. We go beyond simply supporting customers; we build the foundational infrastructure that turns every support interaction into a moment of genuine connection. Whether supporting our established North American community or welcoming new users worldwide, we ensure a unified, reliable, and efficient experience for riders, drivers, applicants and support agents alike. We are looking for an experienced technical leader who can support our global ambition by designing, owning, and scaling the Global Support Platform. Responsibilities: Set the technical vision and strategy for a rapidly evolving product area, making high-judgment calls on architecture and technology direction Drive the adoption of AI and machine learning solutions to optimize customer support workflows and enhance operational efficiency across the platform. Lead a team of talented engineers who ship code and tackle hard engineering problems, maintaining a high bar for technical excellence Translate high-level business goals into actionable engineering projects. Own the technical roadmap from conception to delivery, managing cross-team dependencies and mitigating risks. Drive the responsible adoption of AI development tools across engineering teams - modeling effective use, establishing best practices, and mentoring engineers to improve productivity without compromising code quality or security. Champion improvements in system, observability, performance, and tech debt reduction, extending your influence beyond your immediate team. Establish best practices f

Staff Software Engineer - Testing & Automation Exceptional software engineering is challenging. Amplifying it to ensure that multiple teams can concurrently create and manage a vast, intricate product escalates the complexity. As a Staff Engineer within the Verification Platform team at Sumo Logic, you will drive the implementation and optimization for our verification platform as well as the modernization of our CI/CD pipelines. Your mission is to develop and sustain automated tooling for all testing, verification, and functional requirements, leveraging AI reasoning and machine learning models to predict and prevent delivery issues, while integrating advanced security validation and non-functional requirements into our delivery lifecycle. You will contribute significantly to establishing automated delivery pipelines, empowering autonomous teams to create independently deployable services, and progressing Sumo Logic’s internal Platform-as-a-Service. This role sits at the intersection of Platform Engineering, Quality Engineering, DevSecOps, and Developer Productivity, helping teams deliver secure, reliable, and independently deployable services at scale. Responsibilities Strategy & Leadership: Drive technical direction and design for a modern Quality Engineering platform, driving the adoption of AI reasoning for enhanced automation of all testing, verification, and functional requirements. Pipeline Modernization: Lead the modernization of CI/CD pipelines to include automated security validation, compliance checks, and other critical non-functional requirements, with a focus on integrating AI/ML for intelligent pipeline optimization and risk prediction. Framework Ownership: Own the delivery pipeline and release automation framework for all Sumo services, ensuring improvements in developer productivity, deployment frequency, and release reliability. Cross-Team Collaboration: Educate and collaborate with teams during design and development phases to ensur

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. At Okta, we are building the future of secure, enterprise-grade cloud automation and system connectivity. We are looking for a Senior Software Engineer to join our global Automation Engineering team to design, scale, and govern our enterprise integration substrate using AWS cloud services and modern iPaaS platforms. This is a senior individual contributor role for a hands-on system engineer who can set technical standards, establish integration best practices, and partner with cross-functional teams to automate complex business workflows at scale. What You'll do : Lead technical design and execution for automation initiatives within the team, creating paved paths that enable builders across Okta to connect enterprise systems seamlessly. Design, build, and deploy high-throughput event-driven integration flows , API gateways, and async orchestration workflows using AWS architectures and iPaaS platforms. Build reusable frameworks , developer SDKs, self-service primitives, and integration templates to streamline automation delivery. Serve as a technical domain expert on AWS cloud services and modern iPaaS tooling, driving scalable architecture, reliability, and builder enablement. Partner with operations, security, and platform teams to strengthen monitoring, observability, structured audit logging, and automated governance. Mentor team members and participate in cross-functional design reviews , instilling a platform engineering mindset and raising the b

At Lyft, our purpose is to serve and connect. We aim to achieve this by cultivating a work environment where all team members belong and have the opportunity to thrive. Every day, millions of riders and drivers depend on Lyft to get where they're going. When something goes wrong along the way, they expect us to make it right — quickly, clearly, and without friction. How fast and how well we resolve those moments shapes whether people keep choosing Lyft. The Self-Serve Intelligence team, within the Safety & Customer Care org, is composed of engineers building the AI-powered systems that do exactly that: resolving rider and driver suboptimal experiences without agent involvement through AI Assist (e.g. AI Agents), automations, and self-serve workflows. Our goal is to make getting help feel effortless. We design and build backend services, APIs, and GenAI-powered products that combine robust engineering with applied AI to deliver reliable, scalable self-serve experiences. We are looking for a highly motivated, collaborative, team-focused and technically strong Software Engineer to join our Self-Serve Intelligence team. As a member of this team, you will build the services and AI-powered products that resolve customer issues autonomously. Every day, you'll partner with machine learning engineers, product, design, data science, and operations on high-impact projects — from shipping new AI Agent capabilities, to building the evaluation pipelines that keep their quality high, to improving the backend services underneath them. You'll bring strong engineering instincts, genuine curiosity about applied AI, and a willingness to work through ambiguity in a space that changes month to month. Responsibilities: Write well-crafted, well-tested, readable, and maintainable code Partner with senior engineers to design, build, and ship backend services and GenAI-powered products (e.g. AI Agents) that resolve rider and driver suboptimal experiences Independently lead tasks fro

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta is The World’s Identity Company. We free everyone to safely use any technology anywhere, on any device or app. Our Workforce and Customer Identity Clouds enable secure yet flexible access, authentication, and automation that transforms how people move through the digital world, putting Identity at the heart of business security and growth. At Okta, we celebratea a variety of perspectives and experiences. We are not looking for someone who checks every single box—we’re looking for lifelong learners and people who can make us better with their unique experiences. We are seeking an independent Staff Software Engineer who effectively balances technical excellence with a disciplined approach to the software development lifecycle. What you'll be doing Design, build, and evolve intelligent customer experience solutions that integrate Salesforce CRM with AWS services and modern generative AI capabilities. Architect and develop AI-powered experiences using Amazon Bedrock, LangChain, and modern LLM integration patterns to create context-aware, scalable, and customer-facing solutions. Build robust backend services and integrations to support customer workflows, including LLM orchestration, prompt workflows, RAG pipelines, and other generative AI features. Lead the design and delivery of complex customer experience initiatives from technical discovery through production deployment, ensuring scalability, reliability, and maintainability. Develop and refine prompt e

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. 🚀 The Mission The Foundry is ClickUp's internal AI innovation lab — embedded inside GTM Systems and accountable for turning AI capabilities into production-grade, internally deployed products that make every GTM function faster and smarter. We build the infrastructure that powers AI-first work across Sales, Marketing, Post-Sales, and Revenue Operations. As the Senior Software Engineer on this team you will own the technical delivery of our MCP server platform, agent orchestration layer, and internal tooling — shipping production systems used daily by hundreds of ClickUp employees, and scaling your own throughput by treating AI tools as first-class engineering collaborators. What You'll Own MCP Server Platform Design, build, and operate Model Context Protocol servers that expose CRM, ticketing, analytics, and communication data to AI agents across the GTM stack Implement Okta PKCE authentication flows and RBAC policy enforcement so agents access only the data they're authorized to touch Maintain deployment infrastructure on AWS (Bedrock, Lambda, ECS, API Gateway) and contribute to GCP workloads where applicable Own observability: structured logging, distributed tracing, latency SLOs, and on-call runbooks for every production server Agent Orchestration & AI-Native Products Build and maintain multi-step autonomous agents that execute end-to-end GTM workflows — lead qualification, deal room assembly, onboarding automation, support triage, and more Architect prompt engineering frameworks, tool-call schemas, and agent evaluation harnesses that make AI behavior predictable and auditable Integrate with LLM p

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. We are seeking an experienced and technically influential Senior Software Development Engineer to join our Cloud Tooling and Pipelines team. This pivotal team is responsible for the design, development, and maintenance of our core Continuous Delivery (CD) platform (leveraging Spinnaker and custom tooling), Infrastructure as Code (IaC) execution engines (primarily Terraform), and a suite of supporting microservices. These systems are critical for enabling and managing our extensive resource footprint across AWS ECS and EKS . As a Senior Software Development Engineer, you will be a key contributor, driving the implementation of scalable, reliable, and secure software solutions that automate infrastructure provisioning and application deployments. Your deep expertise in software engineering principles and cloud-native development will be essential in building and enhancing our critical tooling for infrastructure provisioning, vulnerability management, and IaC deployments. You will also play a vital role in mentoring other engineers and influencing the team's technical roadmap. If you have a strong passion for building robust software systems that empower operational efficiency at scale, we encourage you to apply. Key Responsibilities Design and Develop Core Platform Components: Lead the design and development of scalable and reliable microservices and tools that form the backbone of Okta's Continuous Delivery (CD) platform (including components for Spinnaker,

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. The Streaming Foundations team builds services and operates data pipeline infrastructure to support event streaming, messaging, and analytics use cases. We are looking for a Software Engineer who is passionate about distributed systems, platform engineering, and solving data-intensive problems at scale. In this high-impact role, you will get to work with engineers throughout the organization to build foundational infrastructure that allows Auth0 to scale for years to come. What you’ll be doing Help set the technical direction for the team and influence the engineering roadmap for the Platform’s streaming capabilities Design and lead the implementation of our most complex and critical systems for data-intensive use cases. Research and champion new technologies and architectural patterns to solve strategic challenges and scale the platform. Lead and influence cross-functional initiatives, ensuring technical alignment and successful execution across multiple teams. Improve the operational posture of our systems by designing for observability, reliability, and scalability, and by mentoring others in operational best practices. Coach and mentor senior engineers and act as a technical leader across the engineering organization. Collaborate with different stakeholders like product teams whenever needed. What you’ll bring to our teams 7+ years of software development experience in a fast-paced, agile environment Experience working with Golang or Java is preferred H

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. We are seeking an experienced and technically influential Senior Software Development Engineer to join our Cloud Tooling and Pipelines team. This pivotal team is responsible for the design, development, and maintenance of our core Continuous Delivery (CD) platform (leveraging Spinnaker and custom tooling), Infrastructure as Code (IaC) execution engines (primarily Terraform), and a suite of supporting microservices. These systems are critical for enabling and managing our extensive resource footprint across AWS ECS and EKS. As a Senior Software Development Engineer, you will be a key contributor, driving the implementation of scalable, reliable, and secure software solutions that automate infrastructure provisioning and application deployments. Your deep expertise in software engineering principles and cloud-native development will be essential in building and enhancing our critical tooling for infrastructure provisioning, vulnerability management, and IaC deployments. You will also play a vital role in mentoring other engineers and influencing the team's technical roadmap. If you have a strong passion for building robust software systems that empower operational efficiency at scale, we encourage you to apply. Key Responsibilities Design and Develop Core Platform Components: Lead the design and development of scalable and reliable microservices and tools that form the backbone of Okta's Continuous Delivery (CD) platform (including components for Spinna

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. At Okta, we’re building the next generation of authentication for the Agentic AI era. We’re looking for a Senior Engineer to join our AI Authentication team – a group of product-minded, deeply technical engineers delivering features to define what identity and access mean in a world increasingly powered by artificial intelligence. This is a great opportunity for an Senior engineer who thrives in collaborative environments, is eager to grow their technical depth, and wants to build scalable systems that solve real-world security problems. Auth0 Emerging Tech is the Engineering organization where we take care of the hottest technology out there: we ship fast, we don't break things. We are a dynamic and collaborative distributed and diverse team. We value ownership, learning and innovation. We launched our Auth for GenAI offering and we are looking for an Senior Engineer to join us to lead the team taking care of the Identity Protocols parts of the game. Auth0 works with NodeJS ( Javascript or Typescript), a hint of Go and MongoDB or PostgreSQL databases. What will you do: Bring expertise in identity and security while building innovative features and standards that will secure the Agentic AI world Build and maintain scalable services using TypeScript, NodeJS PostgreSQL/MongoDB. Collaborate with other product managers, designers, and senior engineers to deliver features that improve identity and access for GenA

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Required Skills: 9+ years in Python and its libraries (e.g., Pandas, Boto3) for data manipulation, ETL processes, and developing serverless functions to interact with foundational AI services on platforms like AWS Bedrock . 6+ years with modern front-end frameworks (e.g., React, Next.js, TypeScript), and an ability to collaborate effectively with Product & Design. Deep hands-on experience with AWS solutioning , including designing and deploying production applications using core services such as AWS EC2, AWS Lambda, Amazon S3, Amazon RDS, and API Gateway . Familiarity with Infrastructure as Code (IaC) tools like CloudFormation or Terraform is essential. Proven experience building and deploying production-grade AI applications and services. Experience with agentic frameworks like LangChain, LangGraph, and LangSmith is a strong plus. Strong background in building distributed systems, microservices, and resilient APIs (REST/GraphQL) on cloud platforms (AWS/GCP/Azure). Demonstrated ability to influence technical strategy and lead architecture decisions across global teams. Excellent communication skills with experience working effectively across time zones and cultures. Working experience with foundational AI services, with a desire to use a unified platform like AWS Bedrock for Generative AI development and deployment Education and Certifications A Bachelor’s degree in Computer Science, Information Systems, or equivalent years of industry experience AWS cl