Senior AI Engineering Jobs

Wayve is building the AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the Performance Architecture team, which focuses on making Wayve's AI workloads faster and more efficient across training and cloud inference. The goal is to unlock new product capabilities by enabling the training of larger models faster and running inference more efficiently at scale. The team plays a crucial role in the company's mission to make autonomy universal and accelerate the transition to a safer, more intelligent future of mobility.

$370k - $395k

Toast

The Product-Led Growth team at Toast builds in-product experiences that guide restaurants in discovering, trying, buying, and adopting necessary Toast products. This role involves owning the services and surfaces responsible for targeted offers, purchase flows, and activation, with a strong focus on measurable, revenue-attached outcomes. You will ship experiments, analyze results, and develop shared infrastructure to accelerate future launches. This position requires collaboration with product, marketing, sales operations, and data science to translate customer behavior hypotheses into testable in-product experiences and validate them with data.

Speechmatics is seeking a Senior Machine Learning Engineer to join their Research & Development team, focusing on cutting-edge speech technology. The company is a global leader in Voice AI, aiming to understand every voice through innovation and inclusivity. The Modelling Team trains diverse models, including large self-supervised ones, to enhance speech recognition accuracy and ensure their deployment into production using the latest ML developments and robust engineering practices.

Datadog

The Autonomous Ops & Maintenance (ATOM) Team is developing an innovative platform that leverages artificial intelligence and automated systems to empower engineering groups in safely executing complex operations and software maintenance. Their primary objective is to minimize repetitive toil while delivering the precise visibility, oversight, and trust necessary to run automated solutions in mission-critical environments. This platform acts as an intelligent virtual collaborator, directly offloading routine operational overhead that currently requires manual intervention. We are looking for a Senior Software Engineer to help design and scale our foundational architecture, synthesizing vast technical datasets into rich structured knowledge, engineering AI-driven components capable of analyzing intricate operational workflows, and constructing highly reliable systems to execute automated modifications seamlessly.

You will be part of a distributed agile team building a new SaaS platform that underpins enterprise applications for data analytics and other complex problems. Your role will involve ensuring the framework and products meet established design, implementation, and quality goals. This position requires a strong backend engineering focus within a globally distributed team, contributing to the architecture, development, and deployment of Securiti's core backend platform.

We have an exciting and rewarding opportunity for you to take your software engineering career to the next level. You’ll join a collaborative team where your ideas and skills help shape intelligent, data-driven products that power business innovation. As part of a diverse and inclusive environment, you’ll work alongside talented engineers and data scientists, deepening your expertise in Python and machine learning. If you’re passionate about solving complex problems and advancing technology, you’ll find strong support and growth opportunities here. As a Software Engineer III in Corporate Technology’s AI, ML & Data Platforms team, you will design and deliver secure, stable, and scalable technology solutions that support the firm’s data and artificial intelligence objectives. You will work as a valued member of an agile team, tackling technical challenges across multiple areas and contributing to end-to-end delivery. You will help strengthen engineering practices through thoughtful design, strong coding standards, and continuous improvement, while fostering a culture of inclusion and respect.

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the Generative Simulation team, which advances autonomous driving research by incubating new ideas. As a Machine Learning Engineer, you will develop next-generation world models and planners, such as GAIA, to simulate complex driving environments for faster training, broader testing, and scalable deployment. A major focus is on efficiency and interactivity, requiring models to run thousands of roll-outs per second, support closed-loop agent interaction, and fit within practical compute budgets. You will work at the intersection of machine learning research, multi-modal modeling, and real-world deployment, tackling challenges like deploying autonomous vehicles without real-world data and assessing if synthetic environments can replace physical testing.

We have an exciting and rewarding opportunity for you to take your software engineering career to the next level. At JPMorganChase, we believe in investing in our people — offering you the tools, mentorship, and environment to grow, innovate, and make a real impact. You'll be part of a culture that values diversity, inclusion, and continuous learning, where your contributions directly shape the technology that serves millions of customers worldwide. As a Software Engineer III at JPMorganChase within Corporate Technology, you serve as a seasoned member of an agile team to design and deliver trusted, market-leading technology products in a secure, stable, and scalable way. You are responsible for carrying out critical technology solutions across multiple technical areas within various business functions in support of the firm's business objectives. In this role, you will collaborate across teams to solve complex technical challenges, contribute to engineering communities of practice, and help drive a culture of excellence and innovation.

We are seeking a Technical Lead Manager to lead our Synthetic Data team, a critical component of our autonomous driving research. This role will focus on leveraging generative world models, like GAIA-3, to create a production engine for training-grade synthetic data. The goal is to significantly reduce our reliance on expensive on-road data collection by enabling the generation of diverse driving scenarios, even for future vehicle platforms or geographies where data collection is limited. You will lead a team of machine learning engineers, guiding technical strategy, system design, and the development of capabilities that accelerate our journey to AV2.0.

Build the platforms that make advanced AI practical at scale. In this role, you’ll shape standards, tooling, and reliable inference foundations that help engineering teams move faster with confidence. You’ll work hands-on with modern large language model serving stacks and performance tuning, while partnering closely with platform and product stakeholders. As a Senior Lead Software Engineer, you will lead the design and delivery of secure, stable, and scalable platform capabilities that simplify adoption and day-to-day use. You will set technical direction for tooling and runtime foundations, with a focus on production-grade large language model inference and Kubernetes-based deployment patterns. You will partner across engineering teams to improve reliability, developer experience, and operational outcomes through automation and standards. You will mentor engineers and reinforce inclusive, high-accountability ways of working.

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. The Model Integration and Release team owns the engineering release process for Wayve’s driving models, validating that new architectures, features, or data changes meet quality, safety, and performance standards before they reach customers. This role involves working across ML engineering, AI Platform, evaluation, and CI/CD to protect the model baseline while continually improving how reliably and efficiently models are delivered.

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the AV Core Engineering Team, specifically the Core Model Safety team, which focuses on collision avoidance, model understanding, and robustness under failure. You will join a senior team working with large-scale training and fleet data, collaborating with research, simulation, evaluation, and applied engineering teams. The core of the work involves leading large-scale data enrichment and curation initiatives, developing workflows for mining hard and rare events, investigating model performance in real-world scenarios, and improving the efficiency and reliability of the machine learning operations stack.

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the ML performance team, which optimizes inference for edge accelerators and GPUs, ensuring large transformer-based models run efficiently on low-cost, low-power in-vehicle compute. You will work closely with model, platform, and deployment teams to transform research models into reliable production systems for Wayve's driving product. This is a hands-on position involving ML systems, compilers, runtimes, kernels, and embedded deployment, focusing on identifying bottlenecks, implementing performance improvements, and collaborating with model developers on performance trade-offs.

Capital One is at the forefront of leveraging AI to transform banking, creating responsible and reliable AI systems that enhance customer experiences. The Intelligent Foundations and Experiences (IFX) team is central to this vision, building and deploying proprietary AI solutions that deliver value to millions. This role involves partnering with cross-functional teams to deliver AI-powered products, design and develop AI software components, and invent state-of-the-art foundation model optimization techniques. You will contribute to the technical vision and long-term roadmap of foundational AI systems, setting the technical direction for enterprise-wide AI architecture and championing responsible AI principles.

$245k - $279k

We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible. As a Lead Software Engineer at JPMorganChase within the Consumer & Community Banking, you are an integral part of an agile team that works to enhance, build, and deliver trusted market-leading technology products in a secure, stable, and scalable way. As a core technical contributor, you are responsible for conducting critical technology solutions across multiple technical areas within various business functions in support of the firm’s business objectives.

d-Matrix

d-Matrix is at the forefront of software and hardware innovation, aiming to unleash the potential of generative AI. We foster a culture of respect, collaboration, humility, and direct communication, valuing diverse perspectives to create better solutions. We are seeking passionate individuals driven by execution to join us in shaping the future of AI. As a Staff-level individual contributor on the Software Infrastructure team, you will drive the automation for our CI/CD and release pipeline. This involves diagnosing and fixing pipeline failures, such as identifying real regressions, flaky hardware, or infrastructure issues, and pinpointing the exact merge that caused overnight test failures. You will blend AI techniques with traditional software automation to build highly reliable systems and architect compute and storage solutions across cloud and on-prem environments. Your role will involve partnering with various teams to develop and spread best practices, reducing developer toil and accelerating development velocity.

$155k - $235k

We are seeking a Forward Deployed Engineer to join our AI Enablement team. In this role, you will embed with various business teams across Crunchyroll to understand their workflows, identify opportunities for AI and automation, and build practical AI-powered solutions. You will take problems from initial ambiguity through prototyping, production, and adoption, focusing on delivering measurable improvements in speed, quality, cost, or employee experience. This position requires a blend of strong business acumen, consultative problem-solving, and hands-on engineering to create reliable and scalable AI systems, including agents and workflow automation, that integrate with existing tools and data.

$183k - $229k

Runpod

Runpod is seeking a Senior ML Systems Engineer focused on Inference to lead efforts in making LLM inference the fastest and most cost-efficient on their platform. This role involves owning LLM serving performance end-to-end, measuring, understanding, and improving it across various models, hardware, and workloads. The work directly impacts customer latency and cost, requiring a hands-on approach to identify and fix bottlenecks, and ensure reliable production deployment. The ideal candidate will be passionate about AI infrastructure and contribute to building the platform for the next generation of AI developers.

$150k - $220k

Exa is an applied AI lab building a novel search engine. We focus on massive-scale infrastructure for web crawling, training state-of-the-art embedding models, and designing high-performance vector databases. Our goal is to create perfect search over all the world's information, surpassing current capabilities. This role is critical to our search architecture, with flexibility in project assignments based on individual skills and interests. We are seeking a research engineer to tackle the complex challenges of understanding web page content, including parsing, classification, credibility assessment, and misinformation detection, all at a web scale.

$180k - $350k

Lila Sciences is developing autonomous systems for scientific discovery, and this role focuses on building security into the AI infrastructure from the ground up. As a senior individual contributor on the IT & Security team, you will be responsible for security engineering across Lila's AI and ML systems, encompassing training and inference pipelines, model registries, MCP servers, and agent runtimes. Your work will involve translating the AI security roadmap into actionable controls, driving vulnerability remediation, and mitigating risks throughout the AI lifecycle, including data, models, pipelines, and agentic systems. This is a hands-on role requiring technical expertise in cloud security, supply chain security, and AI-specific threat modeling, with a strong emphasis on cross-functional collaboration with AI Platform, AI Research, ML Operations, Science, and AI Safety teams to ensure security is integrated from the design phase.

$252k - $374k

Elastic

Elastic is seeking a Senior Software Engineer to join the Elasticsearch - Storage Engine team. This globally distributed, fully remote team focuses on developing core storage and query technologies for observability workloads. The role involves designing, implementing, and optimizing Elasticsearch's core storage and query engine, with a focus on improving storage efficiency, advancing query processing and performance, and optimizing data layouts. The company emphasizes empowering individual engineers, code review, consensus-driven decisions, and incremental improvements in a collaborative, asynchronous environment.

$133k - $253k

Faire is a technology wholesale platform connecting independent retailers globally. We are seeking a Staff Engineer to own and evolve our data plumbing, ensuring data is accessible, accurate, and queryable for various teams including Product Engineering, Data Science, and Machine Learning. This hands-on role involves designing and building the next generation of our data ingestion and processing systems, and serving as the go-to expert for data movement at Faire.

$247k - $339k

Join us as we build the next generation of enterprise search infrastructure powering AI agents. Our mission is to evolve from traditional product-specific search systems into a composable, agent-native platform that enables agents and applications to discover, retrieve, reason over, and recommend information across heterogeneous enterprise data sources. As a Senior Staff Engineer, you will be the technical owner for foundational search infrastructure that serves as the backbone of our agentic platform. You will drive the architecture, scalability, and evolution of core indexing and retrieval systems while partnering closely with Search Ranking, ML, Platform, and AI teams to deliver production-grade search capabilities that power agent experiences across the company.

$191k - $334k

Capital One is a leader in using machine learning to create personalized customer experiences and is committed to building world-class applied science and engineering teams to deliver breakthrough product experiences and scalable, high-performance AI infrastructure. The Intelligent Foundations and Experiences (IFX) team is at the center of bringing this vision to life, working with partners across the company to advance AI science and engineering and build proprietary solutions. Our AI models and platforms empower teams across Capital One to enhance their products with AI in responsible and scalable ways, delivering value to millions of customers.

$197k - $225k