PyTorch Jobs

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

SpaceXSI is building Super Intelligence to help humanity understand the universe. Our teams are building the full Super Intelligence stack, from supercomputers to Grok models and a suite of applications. We're creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next. The RL infrastructure team is looking for an engineer to help with low precision RL training and inference.

$180k - $440k

SpaceXSI is building Super Intelligence to help humanity understand the universe. We are seeking exceptional Applied engineers to join a high-priority project that approximately 600 million monthly users use. This is an exciting opportunity for individuals with an engineer or scientist background to apply their skills to recommendation systems, ranking algorithms, search technologies, and many other systems. You’ll work at the intersection of advanced AI development and real-world impact, enhancing the ability to connect users with relevant content, accounts, and experiences.

$180k - $440k

Mercor is seeking an AI Infrastructure Engineer to build and manage the GPU and accelerator platform that powers our AI workloads. This role involves overseeing clusters, schedulers, training and inference serving, and the observability and cost data associated with fleet usage. You will write production code daily and own systems end-to-end, from design to on-call responsibilities. Mercor is a leading AI data company, building the layer between human expertise and frontier models, with a mission to organize human intelligence to power the AI economy. We are a profitable Series C company valued at $10 billion.

$200k - $400k

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities. This is a full time role that can be held from one of our US hubs or remotely in the United States.

$153k - $376k

Figma is seeking PhD interns to join the AI Research team, focusing on the intersection of AI and Design. You will develop innovative AI solutions for design and creative products through fundamental and applied research, contributing to new AI-driven features. The team works on optimizing agentic systems, building robust evaluators, and post-training models, tackling challenging technical problems in AI and creativity. This internship offers the opportunity to publish research in leading AI/ML conferences.

$0k - $0k

Figma is seeking applied scientists with a Machine Learning and Artificial Intelligence background to build AI technologies and enhance Figma products. The role involves driving fundamental and applied research, combining industry best practices with a first-principles approach to design and build AI/ML models and systems. This is a full-time position that can be based in one of our US hubs or remotely within the United States.

$153k - $376k

Databricks

Databricks is seeking an AI Forward Deployed Engineering (AI FDE) professional to join a specialized customer-facing AI team. This role involves delivering professional services to help customers build and productionize AI applications, working cross-functionally with engineering and product teams. The ideal candidate is passionate about customer engagement, collaboration, and staying current with GenAI and LLMOps trends. This position can be remote, but requires periodic on-site work in the Washington, D.C., Maryland, or Virginia metropolitan area.

$182k - $250k

Databricks

Databricks is seeking a PhD GenAI Research Scientist Intern to join their Mosaic AI organization. This role focuses on pushing the frontier of domain adaptation for Large Language Models (LLMs) and AI systems, enabling them to work effectively within custom enterprise domains. You will tackle open research problems related to scaling and automating evaluation, fine-tuning with synthetic data, retrieval augmentation, and efficient inference. The intern will collaborate with the research team on projects involving adapting, improving, and evaluating existing methods, designing novel domain adaptation techniques, and composing multiple methods for efficient post-training.

$0k - $0k

NVIDIA is seeking New College Graduates to join our GPU compute architecture and performance engineering team. This role focuses on pre-silicon modeling, data-supported architecture exploration, and GPU microarchitecture improvement for generative AI workloads, including LLM inference and agentic AI. You will contribute to improving the efficiency and scalability of these models on NVIDIA platforms, gaining hands-on experience with performance modeling, bottleneck analysis, and system-level analysis. This is an opportunity to shape the performance of next-generation AI models in data center and edge environments by developing system-level modeling and correlation solutions.

$124k - $242k

NVIDIA is seeking a research engineer to join the Isaac team, focusing on physical AI for robotics. This role involves developing simulation tools for scientists and companies leveraging machine learning in robotics. You will bridge the gap between Isaac product engineering and NVIDIA Robotics research, transforming research concepts into product features and providing user feedback to the research loop. The team works at the intersection of simulation and machine learning, aiming to create the leading tool for evaluating robot foundation models in simulation, building on NVIDIA's expertise in photorealistic rendering and physics simulation.

$184k - $357k

NVIDIA Research is seeking a PhD researcher to conduct groundbreaking research at the intersection of AI, GPU computing, and Electronic Design Automation (EDA). This role offers the opportunity to shape the future of chip design, which is being transformed by deep learning and GPU acceleration. We are looking for creative and collaborative individuals interested in tackling new challenges and contributing to a world-class technology employer.

$168k - $265k

NVIDIA Research is seeking PhD-level students for a research internship focused on AI hardware/software co-design, hardware accelerator architecture, IC design methodology, and VLSI design. The ideal candidate will possess a broad understanding of machine learning fundamentals, quantization and numerical methods for model optimization, digital VLSI circuits, and advanced VLSI design and verification methodologies, including the application of agentic AI to hardware design. This role involves contributing to cutting-edge research in AI accelerators and VLSI design, exploring innovative numerical methods, and potentially collaborating on research prototype testchips.

$0k - $0k

We are the GPU Communications Libraries and Networking team at NVIDIA, responsible for delivering critical communication libraries like NCCL, NVSHMEM, and UCX for Deep Learning and High-Performance Computing (HPC) applications. These applications demand immense compute power and operate at scales involving tens of thousands of GPUs, connected via high-speed interconnects and networking. Communication performance is paramount, directly impacting end-to-end application performance, especially at massive scales. We are seeking a technical leader to manage our NVSHMEM and UCX libraries, offering an exceptional opportunity to advance the state-of-the-art and develop groundbreaking platforms.

$184k - $357k

NVIDIA is seeking a Senior Software Engineer to focus on AI Inference Performance, advancing innovative LLM and VLM inference on NVIDIA GPU-accelerated systems. This hands-on role involves pushing workloads to practical performance limits, delivering measurable gains in latency, throughput, efficiency, and scale. You will translate performance models and profiler data into working code, collaborating with various teams and contributing improvements to open-source inference engines. Your work will directly impact production deployments and help shape future NVIDIA platforms.

$184k - $357k

Sieve is a multi-modal lab focused on curating high-quality training datasets across video, audio, images, text, and 3D, aiming to solve the bottleneck in the growth of AI applications. As a perception-focused Machine Learning Engineer, you will build systems that transform raw sensor footage into training data crucial for robotics and world-model teams. This role requires a strong background in computer vision, particularly in shipping perception systems for robots, devices, or autonomous vehicles, and applying these skills to massive datasets.

Nebius is building a full-stack AI cloud platform to support developers and enterprises from data and model training through to production deployment. As a Senior Machine Learning Engineer on the Applied AI team, you will own model and endpoint optimization from model artifacts through production deployment. Your work will span model internals, quantization, model compression, speculative decoding, KV-cache optimization, inference engines, and serving architecture to improve latency, throughput, memory efficiency, GPU utilization, and cost per token while preserving model quality and reliability. You will collaborate with kernel and platform engineers to investigate serving problems, compare configurations, and resolve performance and quality regressions, validating improvements through reproducible benchmarks and safe production rollouts.

$195k - $262k

Nebius is building a full-stack AI cloud platform to support developers and enterprises in the global AI economy, from data and model training through to production deployment. The Nebius Token Factory is seeking scientists to tackle frontier LLM and VLM inference bottlenecks by turning them into research questions, conducting rigorous experiments, and translating findings into production capabilities. You will own focused research and optimization projects, write strong code, and produce prototypes, publications, and technical artifacts for engineers to build upon. Your research will cover areas like quantization, distillation, speculative decoding, KV-cache optimization, and model/runtime co-optimization, alongside inference engines and distributed inference architectures. You will evaluate trade-offs across model quality, latency, throughput, memory footprint, and cost per token, and collaborate with engineering teams to bring promising results into production.

$195k - $262k

Nebius is building a full-stack AI cloud platform to support developers and enterprises in AI/ML infrastructure, from data and model training to production deployment. This role focuses on the infrastructure for large-scale AI training and model post-training capabilities, specifically for frontier model improvement. The work involves distributed systems, GPU performance, model training frameworks, RL pipelines, and production engineering. The Senior Machine Learning Engineer will own ML work end-to-end, translating ambiguous goals into concrete experiments, implementing and debugging training and RL recipes, building supporting systems, and delivering measurable improvements in model quality, throughput, and reliability. This is a hands-on role requiring independent debugging of both model behavior and distributed training failures.

$195k - $262k

We are building AI that changes how people capture, understand, and act on conversations and meetings. As Director of AI Research, you will lead research and development across automatic speech recognition (ASR), speaker identification (SID), large language models (LLMs), and audio language models. Your work will shape our AI capabilities, product direction, and long-term research strategy. This is a hands-on leadership role for someone who combines research depth with experience building and deploying AI systems at scale. You will set technical priorities, guide model and infrastructure decisions, and work directly with the team on challenging research and engineering problems. Partnering closely with Product, Engineering, and Data teams, you will turn research advances into reliable product capabilities. You will also build the team, experimentation practices, and technical foundation needed to move quickly while maintaining high standards for quality and performance.

$244k - $313k