Lead Software Engineer Machine Learning Jobs

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. As a Senior Software Engineer, Data on the Mapping team, you will collaborate with our world-class team of engineers, product managers, and scientists to grow and improve the quality of recommended routes and accuracy of our travel time estimations. You will lead the architecture and long-term technical direction of our offline experimentation tooling and route simulation services — the systems that let Lyft test routing changes safely before they reach production. You'll also build scalable data pipelines for experimentation, analytics, and machine learning models, along with the data governance and observability systems that keep them trustworthy. Your work will enable integration with partner teams and allow stakeholders across Engineering, Data Science, and Product to make data-informed decisions that directly impact Lyft’s growth and profitability. Our technology stack is based on the latest technologies such as AWS, Databricks, Kubernetes and Airflow. You will work with incredibly passionate and talented colleagues from software engineering, machine learning and data science on projects that directly impact millions of riders and drivers. Responsibilities Own core data pipelines end-to-end, building deep subject matter expertise in the systems you manage and defining/managing SLAs for pipelines, services, and datasets to ensure reliability at scale Serve as the technical owner and architectural lead for our offline experimentation platform and route simulation services, setting technical direction, evaluating trade-offs, and ensuring the systems scale with Lyft's routing and mapping ambitions Continuously evolve data models and schemas to meet business and engineering requirements Develop AI tools that support self-service management of data pipelines (ETL) and schema evolution, and perform han

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 seeking a Principal Software Engineer to lead the design and delivery of enterprise-scale Generative AI solutions that power next-generation healthcare experiences. This role goes beyond hands-on coding—you will define technical strategy, establish architectural standards, and guide multiple teams in building secure, scalable, and cost-effective AI platforms across AWS (Bedrock) and Google Cloud (Vertex AI API). You will partner with product, security, compliance, and enterprise architecture teams to ensure solutions meet business objectives, regulatory requirements, and performance goals. The ideal candidate combines deep technical expertise with leadership skills—capable of influencing cross-org architecture decisions, mentoring engineering teams, and driving responsible AI practices in production. Key Responsibilities Lead end-to-end platform delivery of highly scalable, secure AI services and applications leveraging AWS Bedrock (Foundation Models, Knowledge Bases, Agents, Guardrails) and Google Cloud Vertex AI (Gemini via Vertex AI API, Agent Builder, Vector Search, Search & Grounding) Architect and implement Retrieval-Augmented Generation (RAG) solutions, integrating proprietary data from sources like Amazon S3 and Google Cloud Storage/BigQuery, and using Bedrock Knowledge Bases and/or Vertex AI Search & Groundi

About Anyscale: At Anyscale , we're on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. We’re commercializing Ray , a popular open-source project that's creating an ecosystem of libraries for scalable machine learning. Companies like OpenAI , Uber , Spotify , Instacart , Cruise , and many more, have Ray in their tech stacks to accelerate the progress of AI applications out into the real world. With Anyscale, we’re building the best place to run Ray, so that any developer or data scientist can scale an ML application from their laptop to the cluster without needing to be a distributed systems expert. Proud to be backed by Andreessen Horowitz, NEA, and Addition with $250+ million raised to date. About the Role Anyscale is seeking a Staff Software Engineer to lead the technical vision for our Infrastructure team. As a Staff Engineer, you will be responsible for the architectural evolution of our control plane and data plane, ensuring that our "infinite laptop" vision scales to meet the most demanding distributed AI workloads in the world. You will act as a force multiplier, setting the standards for Kubernetes-based cloud-native infrastructure while mentoring engineers and driving cross-functional alignment across the Ray open-source community and our proprietary product teams. Key Responsibilities Architectural Leadership: Define and drive the multi-year technical roadmap for services that orchestrate Ray clusters across diverse cloud and on-premises environments. Systemic Optimization: Lead the design and optimization of high-performance control plane components specifically tailored for large-scale, heterogeneous AI/ML workloads. Platform Reliability: Establish the organization-wide standards for the reliability, scalability, and observability of Anyscale-managed infrastructure. Strategic Integration: Direct the long-term strategy for accelerator integration (GPUs, TPUs) and container management to ens

Discord has a highly engaged community of millions of daily active users who use the platform for many different reasons, but there’s one thing that nearly everyone does: play video games. Discord plays a uniquely important role in the future of gaming, and we are focused on making it easier and more fun for people to hang out before, during, and after playing games. At Discord, we believe everyone can find a place where they belong. Our mission is to help make it easy for everyone to find and join meaningful conversations and to make every part of our product feel smart and delightful. Discord is a rapidly scaling company that puts Data at the heart of its decision-making and future growth. The Experimentation team at Discord is responsible for building the tools, systems, and processes that empower Discord employees to continually test hypotheses and improve our product in a data-informed way. We are looking for a Staff Software Engineer to lead the development of Discord's next-generation experimentation platform, including building state-of-the-art AI assisted experimentation workflows. You will lead and mentor engineers, fostering their growth and enhancing their skills. What you'll be doing Design, build, and support Discord's next generation experimentation platform: scope of work would include core services, data pipelines, configuration tooling, statistical analysis tools and UX. Collaborate with external (sometimes non-technical) stakeholders and customers throughout the company Understand and influence product telemetry practices to support experimentation needs. Use your technical expertise to build delightful user experience and high-scale data systems that power internal teams and product features used by many millions of users every day Deliver business results by collaborating with stakeholders across Discord to experiment on features, machine learning models and more. What you should have You have 7+ years of experience as a full-stack SWE You have

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 Workflows is the secure, no-code automation platform that empowers organizations to build identity-centric workflows across cloud applications — all without writing code. Our intuitive drag-and-drop interface allows enterprises to automate complex business processes at scale, enhancing productivity, enforcing security, and simplifying IT operations. Customers like Netflix, MGM, and NTT rely on Workflows to automate high-impact identity scenarios with speed and confidence. As we continue to scale, we’re investing in extensibility, developer experience, and performance. Join us to help shape the future of cloud automation and low-code development. Position Description We’re hiring a Staff Full-Stack Engineer to join the Integration Builder team within Okta Workflows. This team owns the core no-code surface that enables both internal teams and third-party developers (ISVs) to build powerful integrations and automation experiences with ease. As a Staff Engineer, you’ll lead initiatives that span front-end and back-end services — delivering performant, secure, and scalable features. You’ll help define architecture, drive implementation, and collaborate closely with Design, PM, and Platform teams. You’ll also work directly with our technical architects to help shape what we build — and how we build it. This is a high-impact role in a growing, strategic product area with strong executive visibility. Role Details: Design, build, and maintain end-to-end feature

Who We Are HP IQ is HP’s new AI innovation lab. Combining startup agility with HP’s global scale, we’re building intelligent technologies that redefine how the world works, creates, and collaborates. We’re assembling a diverse, world-class team—engineers, designers, researchers, and product minds—focused on creating an intelligent ecosystem across HP’s portfolio. Together, we’re developing intuitive, adaptive solutions that spark creativity, boost productivity, and make collaboration seamless. We create breakthrough solutions that make complex tasks feel effortless, teamwork more natural, and ideas more impactful—always with a human-centric mindset. By embedding AI advancements into every HP product and service, we’re expanding what’s possible for individuals, organisations, and the future of work. Join us as we reinvent work, so people everywhere can do their best work. About The Role As the Senior Software Engineer, Tooling and Development Infrastructure, you will play a critical role in shaping the developer productivity tools and automated testing strategy. You’ll collaborate closely with design, development, and quality teams to plan, design, and implement robust automated tools and services that ensure the quality and reliability of our AI software stack. You will be highly hands-on in your work and collaborate closely with stakeholders. This position offers a unique opportunity to influence the development of cutting-edge automation frameworks, foster a culture of quality, and contribute to the long-term success of the organization. What You Might Do Develop and implement automation frameworks and testing strategies that cover the entire software stack, from backend systems to user-facing features. Identify, evaluate, and integrate new tools that streamline development. This includes everything from code quality tools and to Infrastructure-as-Code (IaC) solutions. Lead continuous improvement efforts for our build, release, and test systems, ensuring a robust

About Flexport: At Flexport, we believe global trade can move the human race forward. That’s why it’s our mission to make global commerce so easy there will be more of it. We’re shaping the future of a $10T industry with solutions powered by innovative technology and exceptional people. Today, companies of all sizes—from emerging brands to Fortune 500s—use Flexport technology to move more than $19B of merchandise across 112 countries a year. The recent global supply chain crisis has put Flexport center stage as we continue to play a pivotal role in how goods move around the world. We are proud to have the support of the best investors in the game who believe in our mission, solutions and people. Ready to tackle global challenges that impact business, society, and the environment? Come join us. The opportunity: Flexport is a platform for global trade in an industry that comprises 12% of the global GDP. We are the first licensed freight forwarder and customs brokerage built around a modern tech stack. We help the world's fastest-growing brands manage their international supply chains, providing a fully-managed service for moving products between any two places on the planet. Exceptional software development is integral in allowing us to fulfill our mission of fixing the user experience in global trade. At Flexport you’ll develop products that are at the forefront of reshaping the entire logistics & supply chain industries. You’ll work alongside self-starters interested in solving real-world problems and streamlining the inefficiencies in the complex global trade industry. You’ll have the opportunity to reshape an industry by creating the new operating system for global trade. Our engineers are top-notch software developers who love listening to the users, working as a team, and proactively taking the lead whenever necessary. The role involves learning the business of freight forwarding, modeling the work that people do to move freight, desi

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. Company Description Okta is the leading independent provider of enterprise identity. The Okta Identity Cloud enables organizations to securely connect the right people to the right technologies at the right time. With over 6,500 pre-built integrations, Okta customers can easily and securely use the best technologies for their business. Over 7,950 organizations — including JetBlue, Nordstrom, Slack, and Twilio — trust Okta to protect the identities of their workforces and customers. Position Description We are looking for an experienced Senior Software Engineer – UI to join our Identity Platform engineering team. You will own the design and delivery of complex, enterprise-grade frontend experiences that power Okta's identity lifecycle management capabilities — including admin configuration flows, wizard UIs, real-time progress tracking, and bulk operation workflows. You will partner closely with Product Management, UX, and backend engineers to translate complex enterprise identity requirements into intuitive, accessible, and performant web applications. This is a hybrid position. Job Duties and Responsibilities - Frontend Ownership: Independently own and deliver complex UI features end-to-end — from design collaboration through production deployment. - Architecture & Standards: Lead frontend architectural decisions, enforce code quality, accessibility, perfor

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. At Baseten, we are building the global operating system for distributed, heterogeneous AI hardware. We believe that as LLM and multi-modal workloads scale, the network is the computer. We are looking for foundational engineers to lead our GPU Networking efforts, making RDMA a first-class building block in our infrastructure and unlocking the next generation of distributed inference optimizations. THE OPPORTUNITY Networking and compute are no longer separate disciplines; they are converging. The massive throughput of H100, B200, and NVL72 architectures enables and demands a new approach where communication is co-optimized alongside computation. We are entering an era where the network is an active accelerator, leveraging smart hardware offloads and direct interconnects to ensure that data movement operates at wire-speed. In this role, you will go beyond network configuration to architect the software fabric that unifies thousands of GPUs into a cohesive operating system. While you will leverage the best of the open-source ecosystem, you won't be limited by it. Where off-the-shelf solutions stop, you will build from scratch, engineering the primitives required to co-optimize communication and compute for Disaggregated Serving, Wide Expert Parallelism (WideEP), and lightening cold starts. WHAT YOU'LL DO Make RDMA First-Class: You will work on integrating RDMA/RoCE/InfiniBand capabilities directly into our inference stack,

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. The Data Engineering team at Roblox plays a crucial role in enabling the company's success by developing and maintaining highly leveraged Core Data Sets, frameworks, and tooling to support the growing demand for analytics. As a Principal Data Engineer, you will work to define the data ontology for all of Roblox, establish best practices and standards for data operations and lifecycle management, design and build analytics tooling and frameworks, and influence event instrumentation. Additionally, this role is highly cross-functional, requiring close collaboration with Data Science, Experimentation, and Machine Learning teams to understand customer requirements and analytics applications, as well as with Data Infrastructure and Storage teams to develop integrated solutions. Join us and be a part of a dynamic team driving innovation and growth at Roblox. You Will: Partner with Data Science, Data Platform, Product, and Engineering to collect requirements to define the data ontology for all of Roblox Lead and mentor a growing team of Data Engineers to support Roblox's ever-evolving data needs Design, build, and maintain efficient and reliable batch and streaming data pipelines to mod

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. The Data Engineering team at Roblox plays a fundamental role in enabling the company's success by developing and maintaining highly leveraged Core Data Sets, frameworks, and tooling to support the growing demand for analytics. As a Senior Data Engineer, you will work to define the data ontology for all of Roblox, establish standard methodologies for data operations and lifecycle management, design and build analytics tooling and frameworks, and influence event instrumentation. Additionally, this role is highly multi-functional, requiring close collaboration with Data Science, Experimentation, and Machine Learning teams to understand customer requirements and analytics applications, as well as with Data Infrastructure and Storage teams to develop integrated solutions. Join us and be a part of a dynamic team driving innovation and growth at Roblox. This role will report to our Engineering Manager on the Data Engineering team. You Will: Partner with Data Science, Product, and Engineering to collect requirements to define the data ontology for all of Roblox Lead and mentor a growing team of Data Engineers to support Roblox's ever-evolving data needs Design, build, and maintain efficient an

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. What You’ll Do: The Marketplace team, part of the Economy org, owns the platform and infrastructure that supports the Avatar customization and Avatar Marketplace within Roblox. As a Senior Engineer on our team, you will lead the work on building personalized and streamlined shopping experiences for over 100M users within Roblox. Beyond this scope, you will be involved with several additional areas also owned by subteams such as: Avatar Identity Scaling and Serving Avatars to tens of millions of users on the platform Building new Avatar customization features and continuously improving user experience Driving users and creators towards high-quality, high-fidelity Avatars that will shape Roblox’s look and feel going forward The systems above power the engine APIs that enable developers to monetize their games on Roblox, earning hundreds of millions of dollars per year, while processing billions of transactions. The ideal candidate is passionate about understanding customer needs, optimizing performance, and delivering exceptional user experiences. You Have: 3+ years of professional experience in software development Proficiency in one or more programming languages such as C#, Go,

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. Get to know Okta Team Okta is The World’s Identity Company. We enable everyone to safely use any technology anywhere, on any device or application. Our Workforce and Customer Identity Clouds provide secure, flexible access, authentication, and automation, transforming how people navigate the digital world and placing Identity at the core of business security and growth. At Okta, we are committed to celebrating a variety of perspectives and experiences. We are looking for lifelong learners and individuals whose unique experiences will make our organization better. The Technology Data and Intelligence (TDI) team at Okta is focused on boosting internal efficiency through the implementation of secure, scalable, and innovative systems. About the Role We are seeking an independent Senior Salesforce Engineer who effectively balances technical excellence with a disciplined approach to the software development lifecycle. In this role, you will serve as a guardian of our codebase through implementation, rigorous peer reviews, automated testing, and comprehensive deployment support, ensuring the architectural integrity of our environments. Key Responsibilities Technical Implementation: Lead the translation of user stories into robust, scalable technical designs. Responsible for the end-to-end execution of features, ensuring alignment with organizational standards. Solution Design and Architecture: Architect and design secure, scalable solutions by translating complex

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 Identity Security Posture Management (ISPM) Team At ISPM (formerly Spera Security), we are revolutionizing identity security for modern organizations. Our platform provides security leaders with end-to-end contextual visibility into their identity landscape—reducing risk, enhancing operational efficiency, and ensuring continuous compliance. We’re a fast-moving, high-impact team looking for an exceptional Staff Software Engineer to help shape the future of identity security and platform architecture. Hybrid work - 3 days a week from the office. What You’ll Be Doing As a Staff Software Engineer, you will serve as a technical authority and play a pivotal role in scaling our core platform. Your focus will span architecture, security research, and engineering excellence: Lead & Architect: Design, build, and scale high-performance software while driving architectural best practices across the R&D organization. Security & AI Innovation: Drive deep-dive research into cloud, SaaS ecosystems, and AI agents to build proactive detection models, mitigation strategies, and threat intelligence. Engineering Excellence: Elevate code quality, testing standards, system monitoring, and documentation across teams. Cross-Functional Ownership: Partner closely with Product and R&D to translate complex security requirements into scalable technical solutions. Scale & Optimize: Identify and resolve systemic performance bottlenecks and architectural challenges.

Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work together in real time from anywhere in the world. If you're excited to shape the future of design and collaboration, join us! As Figma’s customer base continues to expand and diversify, and new technological advancements in AI and generative design emerge, the opportunity for AI experience engineering is greater than ever. You’ll collaborate closely with a cutting edge AI Research team and your work will help raise the ceiling of what’s possible for designers and lower the floor of who can design. We are seeking an accomplished and experienced AI Product Engineer who has built generative AI products to join our team and lead our AI efforts from inception to implementation. Join us in molding the future of AI at Figma! This is a full-time role that can be undertaken from one of our US hubs or remotely in the United States. What you'll do at Figma: Lead the development of early-phase AI systems to enhance user experience and develop systems to measure and improve AI performance and relevance Direct the future of AI at Figma in collaboration with cross-functional teams of innovative software engineers, product managers, and data scientists Break down open-ended AI problems into engineering strategies utilizing machine learning, data analysis, and experimental design Provide mentorship and promote best practices in data and AI while cultivating a collaborative and inclusive team culture Contribute to the growth of the team by improving the hiring process and recruiting outstanding AI talent We’d love to hear from you if you have: 5+ years of experience of Software Engineering, with 1 or more years of developing generative