PyTorch Jobs

Databricks

Databricks is seeking an engineer to join a new, nimble team in NYC to build CustomerLake, the Customer Data Platform on Databricks. This platform aims to bring enterprise-grade ML and AI personalization to companies whose data already resides on Databricks, enabling them to power personalization, recommendations, churn, and lifetime-value modeling without an in-house ML/AI team. This is a unique opportunity to work in a 0-to-1 environment, combining startup excitement with the resources of a major tech company.

$190k - $253k

Growth Platform at Stripe builds machine learning systems to help businesses discover and utilize Stripe products. These recommendations are integrated across various platforms including the Dashboard, email, onboarding, documentation, and AI agent interfaces. The team combines business understanding with models to determine the most useful recommendations, optimal timing, and effective learning from outcomes. Their work encompasses recommendation and ranking models, contextual bandits, agent-based recommendations, and the underlying data and evaluation systems, aiming to drive business growth through relevant product adoption while maintaining a positive user experience.

The Radar ML team builds the fraud detection models that protect Stripe's $1.9 trillion payment network from fraud. The team owns over 10 real-time deep learning models that must constantly evolve to stay ahead of fraudsters. Each ML improvement translates directly into dollar impact for Stripe and its users. The team's models also power the Radar product suite that tens of thousands of businesses use to screen payments and manage fraud. In this role, you will own ML work across the full lifecycle: researching new fraud patterns, building and deploying models, and sharing results directly with top Stripe customers. You will have opportunities to optimize Stripe’s most intensive ML models, and opportunities to ship 0-to-1 products from scratch.

Financial Connections is Stripe's open banking platform, enabling businesses to securely access consumer-permissioned financial data. Our platform connects to thousands of financial institutions, powering use cases from account verification to risk assessment to personal financial management. Across the Financial Connections Engineering org, we focus on delivering high-quality, enriched bank data at scale — building the ML systems that transform raw financial data into actionable signals for both internal Stripe teams and external merchants. Our ML work spans transaction categorization, risk scoring, data enrichment, and the development of intelligent systems that improve data quality across our network. We operate at the intersection of fintech infrastructure and applied machine learning, solving problems that directly impact Stripe's ability to serve millions of businesses and consumers.

SpaceXSI is building Super Intelligence to help humanity understand the universe, and you will join the multimodal team to push toward superhuman multimodal intelligence. This role involves advancing understanding and generation across modalities—image, video, audio, and text—spanning the full stack from data curation to end-to-end product experiences. You will collaborate cross-functionally to deliver frontier capabilities in multimodal reasoning, world modeling, tool use, agentic behaviors, and interactive human-AI collaboration, contributing to models that can see, hear, reason about, and interact with the world in real time.

$180k - $440k

NVIDIA is seeking a Software Engineer, Performance Instrumentation to join their team in Santa Clara, CA. This role offers an exciting opportunity to contribute to groundbreaking projects at the intersection of AI and accelerated computing. You will be instrumental in defining and maintaining performance data requirements, identifying and addressing visibility gaps, and ensuring the quality and efficiency of instrumentation APIs. This position requires a strong background in software engineering with a focus on performance and data quality within machine learning or systems software environments.

$168k - $270k

NVIDIA is seeking a dedicated Software Engineer, ML Performance to join our team in Santa Clara, CA. In this role, you will engage directly with the latest AI technologies and work on optimizing performance at an outstanding scale. Your expertise will ensure that our groundbreaking solutions operate flawlessly and continue to lead the industry. You will be immersed in an inclusive, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

$224k - $357k

Glean is seeking Machine Learning Engineers to join the Enterprise Brain team, focusing on building proactive AI products that revolutionize enterprise workflows by automating tasks and enhancing user productivity. This role involves leveraging deep user understanding and state-of-the-art enterprise graphs, utilizing LLMs, advanced ML techniques, agent orchestration, and cutting-edge ranking methods. You will contribute to a secure, open platform that redefines how enterprises find, use, and act on knowledge, with a focus on enterprise trust and scalability.

$200k - $280k

Crusoe is seeking a highly skilled Software Engineer to join their Data Center Infrastructure Engineering team. This role focuses on developing software for managing a fleet of GPU servers and the data centers that house them. The position involves creating and implementing advanced diagnostic, observability, automation, and repair tooling for high-performance GPU compute clusters. The ideal candidate will be a hands-on problem solver comfortable working independently, playing a critical role in maintaining the health and scalability of Crusoe's rapidly growing GPU fleet.

$170k - $205k

Anduril Industries is seeking a High Performance Computing (HPC) System Engineer to directly support sensitive programs by building and maintaining large-scale HPC infrastructure. This role offers the opportunity to work with leading engineers and cybersecurity professionals on cutting-edge systems deployed globally for national security missions. You will be instrumental in architecting, deploying, and managing advanced GPU infrastructure, ensuring the seamless operation of critical compute environments.

$147k - $196k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence for home and commercial markets. The Helix team is responsible for developing the core AI systems that power humanoid autonomy. We are seeking a Helix AI Engineer, Video Pretraining to lead the development of large-scale video foundation models trained on diverse real-world and robot-collected data. This role focuses on pretraining models that learn from raw video, capturing motion, interaction, and temporal structure to enable downstream capabilities in perception, prediction, and embodied reasoning.

$200k - $400k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping human-level intelligence. The Helix team is seeking an experienced AI Training Performance Engineer to elevate model training. This role focuses on enhancing distributed training frameworks for large-scale model training, optimizing GPU kernels, evaluating different accelerator types, and co-designing models for maximum hardware utilization.

$200k - $400k

SambaNova is seeking a new grad ML Infrastructure Engineer to join their team. This role involves writing production Python code for the inference serving stack, including vLLM integration and public APIs. You will contribute to bringing features like speculative decoding, prompt caching, and tool calling into production on the RDU, and develop accuracy and regression tests to ensure quality before customer release.

$131k - $197k

Databricks

The AI Forward Deployed Engineering (AI FDE) team at Databricks is a specialized customer-facing AI team focused on building and productionizing first-of-its-kind AI applications. This role is for experienced engineers who excel at working with customers and teammates, and who are passionate about the latest trends in Generative AI, LLMOps, and Machine Learning. You will collaborate cross-functionally with engineering and product teams to shape strategic priorities and initiatives, acting as a trusted technical advisor to customers across various domains.

$182k - $250k

CoreWeave is seeking a Software Engineer to join the Inference team, focusing on shipping production features that enhance model serving on their GPU platform. This role involves improving latency, reliability, and cost-efficiency. As an IC1, you will implement well-scoped changes, learn operational practices, and grow with mentorship from experienced engineers. CoreWeave is the essential cloud for AI, providing a platform of technology, tools, and teams that enable innovators to build and scale AI with confidence.

$92k - $135k

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team. This role will be instrumental in managing a planet-scale performance data warehouse, handling the ingestion, storage, transformation, and analysis of performance events across global data centers. The engineer will also contribute to industry-leading performance benchmarking publications like MLPerf. The position involves owning designs, elevating engineering standards, and driving measurable improvements in latency, throughput, and reliability across multiple services. Collaboration with product, orchestration, and hardware teams will be key to evolving the Kubernetes-native platform and meeting stringent P99 SLAs at scale.

$182k - $242k

CoreWeave is seeking an early member for its Applied Training team to build the research infrastructure that enables AI pioneers to train models with confidence. This role addresses the problem of AI labs spending valuable research time on cluster setup and operations instead of innovation. You will contribute to the roadmap and develop solutions for our Kubernetes-native research cluster platform or sandbox client for agentic training and evaluation, aiming to provide frontier lab-level infrastructure to all CoreWeave customers.

$182k - $242k

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team, focusing on GPU kernel authoring and optimization. In this role, you will be responsible for writing, profiling, and tuning GPU kernels that are critical for large-scale model serving, aiming to maximize throughput and minimize latency. You will also contribute to industry-leading performance benchmarking publications. This position requires an individual who can lead designs, improve engineering standards, and deliver significant enhancements to latency, throughput, and reliability across the inference stack. You will collaborate with product, orchestration, and hardware teams to translate kernel-level optimizations into system-wide improvements and meet stringent Service Level Agreements (SLAs) at scale.

$182k - $242k

CoreWeave is seeking a Senior Engineer to join its Benchmarking & Performance team. This role will focus on building performance insights and observability systems for AI infrastructure, covering everything from individual GPUs and interconnects to distributed training and inference workloads. You will be responsible for detecting, diagnosing, and surfacing performance signals across global data centers, transforming raw telemetry into actionable insights for engineers, product teams, and executives. This is a hands-on role focused on building observability tooling that provides near real-time answers about infrastructure performance, rather than a traditional data engineering or BI position.

$182k - $242k

CoreWeave is seeking a Principal Engineer to lead its Benchmarking & Performance team. This role will be responsible for a global performance data warehouse, ingesting, storing, transforming, and analyzing performance events across all data centers. You will also play a key part in industry-leading performance benchmarking publications, working closely with NVIDIA and the open-source community to demonstrate CoreWeave's performance and reliability leadership in AI.

$206k - $333k