Senior AI Engineering Jobs

As a Director of Software Engineering at JPMorganChase, you will lead a technical area and drive impact across teams, technologies, and projects. You will utilize your in-depth knowledge of software, applications, technical processes, and product management to drive multiple complex projects and initiatives. This role involves serving as a primary decision-maker for your teams and a driver of innovation and solution delivery, focusing on agentic architecture and protocol standards.

Salary not listed

Vercel

As a Senior Forward-Deployed Engineer on the Professional Services team, you will embed with developer teams at strategic enterprise customers to lead frontend migrations, performance work, and production AI builds. Success looks like production code live in customer environments, customer teams able to extend it on their own, and a wider partnership with each account. This is a remote role open to candidates in the US, with occasional travel to customer sites.

$137k - $207k

CoreWeave

CoreWeave is seeking a Staff Software Engineer to join the Inference Platform Team. This team builds and operates CoreWeave’s Kubernetes-native inference platform, designed for low-latency, high-throughput AI workloads at massive scale. As a technical leader, you will drive architecture, performance, and reliability across multiple services and teams. Your responsibilities will include optimizing inference performance (latency, throughput, and GPU utilization), improving system reliability, and working deeply in distributed systems and Kubernetes-based infrastructure, focusing on areas like scheduling, batching, and memory optimization. This role requires hands-on technical leadership and the ability to influence engineering direction across the organization.

$188k - $275k

DigitalOcean is seeking a Staff Engineer to design and optimize their Serverless Inference infrastructure and APIs. This role will focus on tackling large-scale AI workloads, emphasizing throughput, GPU utilization, and fault tolerance to support the next generation of AI-native enterprises. You will be instrumental in building scalable, multi-tenant services for AI inference and intelligent routing, while also strengthening platform resiliency through enhanced observability, capacity management, and automation. This position requires close collaboration with platform, GPU infrastructure, and product engineering teams to deliver robust, highly available systems and APIs.

$191k - $239k

At Klaviyo, we believe the future of software lies in systems that can run and optimize themselves based on outcome or reward metrics. We have built the infrastructure and application that serve as the interface between businesses and consumers, now serving over 200,000+ customers. We have a significant opportunity to develop state-of-the-art AI and machine learning technologies to power our products and create AI agents that can automatically generate and execute marketing strategies and campaigns. As a Senior Machine Learning Engineer, you will help build models that extract insights from massive data streams, applying cutting-edge techniques from deep learning, recommender systems, and language modeling to integrate AI into our product and deliver customer value. You will work closely with partners in Engineering, Product, and Design, and engage directly with stakeholders and customers to solve challenging problems.

$196k - $294k

Torc is seeking a Senior Machine Learning Engineer to join the Ego-Motion & Localization Team. This team is responsible for developing the ML models, state estimation algorithms, and production software that estimate vehicle pose, velocity, and acceleration in real-time. The solutions fuse information from multiple sensors and maintain robust performance under various conditions. As part of this team, you will develop and deploy production-ready localization and sensor fusion solutions, contributing to the challenges of bringing autonomous vehicle technology to commercial applications.

$177k - $213k

NVIDIA is seeking a Senior Developer Relations Manager to drive the adoption of its Cosmos 3 and Cosmos world foundation models within the Robotics ecosystem. This role involves building strong technical relationships with strategic partners and the NVIDIA Cosmos Coalition community, guiding developers in applying these models for robotics and autonomous systems. The ideal candidate will possess deep technical knowledge, adaptability, and expertise in guiding architecture and implementation plans for physical AI, world foundation models, and robot learning.

$184k - $357k

Anduril Industries

Discovery is seeking an AI & Agentic Platform Engineer to join their expeditionary force. This role involves taking unproven concepts and building the analysis or prototype to test them, working on challenges across autonomy, sensor analysis, and battle space management. The team focuses on moving AI models from research into production, specifically onto classified platforms and edge hardware where reliability is paramount. Rigorous and repeatable evaluation of model performance is critical as the model portfolio and classified work expand.

$260k - $345k

Vercel

Vercel is seeking a DevRel Engineer to join their product teams focused on agentic infrastructure, including AI Gateway and Sandbox. In this role, you will work closely with product and engineering to build with these tools before their public release, showcasing their capabilities through various content formats. You will be instrumental in shaping the products by providing early feedback and ensuring a smooth developer experience. This position reports to the Head of AI Infrastructure and can be located anywhere.

$168k - $252k

This role is for a Senior Software Engineering Manager within the Corporate Technology, Asset & Wealth Management Finance team. You will lead multiple engineering teams responsible for delivering data engineering and business intelligence solutions that support finance stakeholders. Your focus will be on setting clear technical direction, establishing operating practices, and defining delivery expectations across teams, ensuring a balance between speed, quality, and operational excellence. You will collaborate closely with business and technology leaders to prioritize outcomes and align roadmaps, fostering an inclusive culture centered on ownership, continuous improvement, and the responsible application of enterprise-authorized AI capabilities. This is an opportunity to combine people leadership with technical depth and achieve measurable impact.

Salary not listed

CoreWeave

CoreWeave is seeking a Staff Software Engineer for its Identity & Access Management (IAM) team. This is a high-impact leadership role focused on architecting and expanding CoreWeave's IAM strategy to secure its high-performance cloud infrastructure. The engineer will balance rapid execution with a long-term technical vision to meet the demands of global enterprise customers, designing and extending a platform that secures the company's rapidly expanding cloud platform.

$188k - $275k

DigitalOcean is seeking a Principal Engineer to lead model optimization efforts for serving across their entire accelerator fleet, including NVIDIA and AMD hardware. This role is critical for bridging the gap between a model that simply runs and one that performs efficiently, directly impacting customer Service Level Objectives (SLOs) and GPU resource utilization. You will be responsible for the end-to-end model optimization process, encompassing quantization strategies, kernel selection and development, efficient execution of attention and Mixture of Experts (MoE) layers, speculative decoding, and parallelism layouts. This work involves adapting to rapidly evolving model architectures and diverse hardware vendor stacks, requiring the development of robust methodologies and tooling to ensure all models perform optimally on all GPUs, enabling rapid deployment of new models.

$250k - $312k

Klaviyo is seeking a Staff Engineer to join the Customer Agent team, a new and ambitious group focused on building a conversational AI system for brands to serve their customers at scale. This role is based in Boston and requires an in-office presence. You will be instrumental in setting the technical and product direction for a core area of the Customer Agent product, taking ownership of complex, ambiguous problems from conception to production. The team operates with the speed and ownership of a startup, backed by the resources of a leading tech company, and emphasizes shipping production-ready AI systems that make real judgment calls. You will be hands-on, contributing to architecture, writing code, and mentoring other engineers through design reviews, while partnering with product, design, and data science leaders to shape the future of the platform.

$176k - $264k

The Scene Generation team at Torc is at the forefront of developing sensor simulation technology using Neural Rendering and generative models. This technology is crucial for Torc's data-driven simulator, enabling the creation of high-quality data from novel scenarios to address the domain data gap in autonomous driving. The team focuses on state-of-the-art approaches for generating realistic camera, LiDAR, and Radar sensor data, which are vital for training and validation across the entire autonomous vehicle stack. As a Staff ML Engineer, you will play a pivotal role in leading this team, defining its technical direction, and managing daily priorities to advance Torc's AV 3.0 strategy.

$250k - $300k

Join NVIDIA's Analytics and Data Intelligence (ADI) group to develop "libcudf," an open-source CUDA C++ library that accelerates database and DataFrame operations. This role involves driving speed-of-light performance in structured data processing across various hardware, from single workstations to multi-node GPU supercomputers. You will build the computational core for DataFrame and database accelerators, creating highly optimized C++ and CUDA libraries that leverage GPU parallelism for operations like data loading, parsing, joins, and aggregations. This is an opportunity to contribute to an open-source software suite and make a significant impact.

$184k - $357k

Anduril Industries

We are seeking a Staff ML Tech Lead Manager to apply the latest research to solve our toughest problems. This role will be responsible for owning the entire machine learning stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for perception on edge compute devices that push the boundaries of what's possible with our sensors. You will develop learning algorithms to optimize the behavior of autonomous systems. Your work will directly impact Anduril's ability to deliver cutting-edge defense technology, with the opportunity to identify and develop novel ML applications across our product portfolio.

$220k - $330k

CoreWeave

CoreWeave is seeking a Staff Software Engineer for the Developer Experience (DevEx) team. This team is responsible for the platform, services, and systems that empower all engineers at CoreWeave to build and ship software efficiently and reliably. The scope includes CI/build infrastructure, testing, delivery systems, artifact management, internal developer platforms, and AI tooling. This role is for a technical leader who will define the technical strategy for DevEx, address complex developer productivity challenges, and shape the future of the agentic developer platform.

$188k - $275k

DigitalOcean is seeking an AI Architect to define the technical vision for how the company operates as an AI-native business. This role will own the architecture for agentic systems and the enterprise architecture they integrate with, enabling any team to deploy governed agents into business workflows quickly and safely. You will be responsible for the end-to-end architecture, including how agents are built and run, their interaction with models and enterprise systems, long-running work orchestration, and the implementation of identity, audit, and observability. This is a hands-on role involving coding, prototyping, and setting technical standards across distributed teams.

$191k - $239k

Sony Interactive Entertainment (SIE) is seeking a Sr. Software Development Engineer in Test specializing in ML/AI quality. This role focuses on developing automation for model evaluation, leveraging LLMs for test generation, and validating AI-powered workflows. You will guide quality strategy, build automation frameworks, and manage the implementation of complex, cross-functional machine learning products and services. The position involves leading quality initiatives from inception to delivery, ensuring team synchronization, managing dependencies, and confident release execution. You will collaborate closely with ML, engineering, product, and infrastructure teams to define and implement quality standards, monitoring, and expansion, while leading embedded Quality Engineering (QE) efforts within projects.

$153k - $229k

The Auto Tagger team is responsible for transforming vast amounts of raw, multi-modal vehicle data into a curated library of critical driving scenarios. By identifying long-tail events in driving logs, the team provides essential data to accelerate the development of safe autonomous trucking. Leveraging Pegasus logical layers, this role will structure and catalog findings into an observations database, directly impacting autonomous perception, sensor fusion, and generative simulation testing.

$161k - $193k

NVIDIA is building a unified foundation model that understands relational databases and heterogeneous graphs, enabling a new era of enterprise AI. This role involves designing, building, and evaluating novel Transformer and graph neural network architectures that generalize across diverse data schemas. The work will power applications like recommendation systems, demand forecasting, fraud detection, and predictive maintenance. You will collaborate with researchers and engineers across the entire machine learning lifecycle, from architecture exploration and large-scale training to post-training optimization and high-performance inference, translating foundational research into production systems.

$184k - $357k

CoreWeave

CoreWeave is seeking a Staff Machine Learning Engineer to join the AI Insights team. This role will be a technical leader responsible for defining and implementing machine learning systems that power anomaly detection, signal correlation, incident understanding, recommendations, and other intelligence capabilities across CoreWeave’s observability and cloud platforms. You will work across the full machine learning lifecycle, from problem framing and model development to evaluation, service building, and scaled operation. The position involves close collaboration with software engineers, product managers, researchers, and domain experts to transform innovative ideas into reliable customer experiences, focusing on building underlying ML systems for AI-powered troubleshooting and optimization in real-world environments.

$180k - $250k

DigitalOcean is building the infrastructure that enables developers to build, run, and scale AI-powered applications and autonomous coding agents. As a Senior Software Engineer on the MARS (Managed Agent Runtime Services) team, you will help build a production-grade platform for running coding agents and agentic workflows securely, reliably, and at scale. You will design and implement core components of our managed agent runtime, including agent framework integrations, execution environments, control-plane APIs, event-driven infrastructure, and developer-facing APIs and tools. You will work across distributed systems, cloud infrastructure, and emerging agentic AI technologies to make it easy for developers to build sophisticated agents without having to manage the underlying infrastructure. We are looking for a strong engineer who combines deep software engineering fundamentals with pragmatic architectural judgment, end-to-end ownership, and a passion for simplifying complex technology for developers.

$167k - $209k

Torc is seeking a Senior Machine Learning Engineer focused on 3D Perception to join their Multi-Modal Perception team. This role is crucial for developing the machine learning systems that enable autonomous trucks to perceive and understand their surroundings. You will advance Bird's Eye View (BEV) perception capabilities, creating production-ready solutions to enhance environmental understanding, model robustness, and overall system performance within Torc's autonomy stack. This is a collaborative role where you'll work with engineers and researchers to tackle complex real-world perception challenges.

$177k - $213k