PyTorch Jobs

Luma is seeking a Research Scientist/Engineer to focus on optimizing the performance of their multimodal models. This role involves profiling and enhancing GPU, CPU, and accelerator code to ensure efficient training and scalable deployment without compromising quality. You will be responsible for writing kernels and operations that maximize hardware utilization and minimize latency, requiring deep expertise in GPU optimization, transformer internals, and low-level programming.

Luma is seeking a Research Scientist / Engineer to build and optimize the distributed systems that train large-scale multimodal models across thousands of GPUs. This role focuses on advanced parallelism, training stability, and utilization across massive clusters, requiring deep expertise in distributed PyTorch and CUDA. The goal is to provide researchers with reliable, efficient, and scalable infrastructure for innovation.

$195k - $395k

Luma is seeking a Research Scientist / Engineer to build and optimize the distributed systems that train large-scale multimodal models across thousands of GPUs. This role focuses on advanced parallelism, training stability, and utilization across massive clusters, requiring deep expertise in distributed PyTorch and CUDA. The goal is to provide researchers with reliable, efficient, and scalable infrastructure for innovation.

Snorkel AI is seeking a Staff Applied AI Engineer to join their frontier AI data lab. In this role, you will research and implement state-of-the-art Generative AI and machine learning techniques to deliver impactful solutions for customers. You will work directly with clients to understand their needs, design AI solutions, and contribute to Snorkel's tooling by translating real-world challenges into reusable recipes and workflows. This position is ideal for individuals who enjoy solving complex problems, bridging AI technology with business value, collaborating with customers, staying current with AI research, and standardizing solutions into internal tooling.

$230k - $360k

Baseten powers mission-critical inference for leading AI companies by uniting applied AI research, flexible infrastructure, and seamless developer tooling. We enable companies at the frontier of AI to bring cutting-edge models into production. This role focuses on the applied end of post-training research, working directly with stakeholders from fast-growing AI companies to post-train open-source models that outperform frontier closed models on specialized tasks. The day-to-day involves extracting signal from complex, domain-specific datasets and building reward functions, environments, eval harnesses, and training pipelines to improve models that ship to production and reach millions of users.

$200k - $275k

Descript's Research team focuses on building specialized models for specific editing workflow problems, aiming to ship them as production features used by millions of creators. This role is centered on multimodal understanding, developing models that perceive edited media like a human video editor. The goal is to enhance Descript's AI editing agent, Underlord, by enabling it to directly perceive media, judge its own output, and reason about creative choices beyond just project structure. This is an open research problem with a unique dataset available.

$197k - $263k

ClickHouse is a rapidly growing cloud company specializing in real-time analytics, data warehousing, observability, and AI workloads. The AI/ML Engineering team is responsible for building and operating the company's AI and machine learning products end-to-end, including the Agentic Data Stack, AI Functions, and in-console copilot. We are seeking experienced software engineers to design, build, and operate these cutting-edge AI products, making ClickHouse the platform of choice for agents and data scientists.

$150k - $238k

Baseten is seeking an Engineering Manager to lead a portion of their Inference Performance team. This team is responsible for accelerating and optimizing the performance of demanding AI workloads on GPUs. The role involves managing and growing a team of inference performance engineers focused on the inference engine and runtime, including areas like kernels, scheduling, batching, KV-cache management, and speculative decoding. It's a hands-on technical leadership position where you will define the team's direction, solve complex problems, and contribute technically to GPU performance while also focusing on hiring, developing, and supporting your team members. Your team's work will directly impact customer model inference speed and serving efficiency, and you will play a key role in shaping the team's structure as it scales.

$240k - $270k

NVIDIA is seeking Senior Deep Learning Scientists to advance their work in streaming and agentic multimodal AI. This role involves developing models capable of reasoning, planning, and acting across diverse modalities, with a focus on defining core algorithmic improvements for multimodal foundation models. You will contribute to high-impact, large language models and multimodal AI products that enhance user experiences, working within the Nemotron LLM team to solve real-world agentic AI challenges.

$184k - $288k

NVIDIA's Autonomous Systems and Physical AI Research (ASPIRE) group is seeking top Research Scientists to advance the field of intelligent machines for autonomous vehicles, robotics, and physical AI. This role focuses on fundamental research challenges, including developing new training paradigms, ensuring safety and reliability in critical applications, enabling navigation in novel environments, and building systems that reason and interact safely with humans and other agents. Researchers are encouraged to publish their work and contribute to open-source projects, collaborating with internal product teams and external researchers to influence real-world products while pursuing groundbreaking research.

$168k - $265k

NVIDIA is seeking a Senior System Simulation Architect to drive the development of CPU architectures for AI, deep learning, high-performance computing, gaming, virtual reality, and autonomous vehicles. This role involves developing and integrating full-system functional models, bringing up system and application software in simulation and emulation, and supporting CPU architects and performance engineers. The position is crucial for pushing performance boundaries for NVIDIA's CPU products and integrating them seamlessly with NVIDIA's broader technology stack to enable faster AI model training, agentic use-cases, efficient data processing, and scalable cloud deployments.

$184k - $357k

Knowtex is seeking an ML Engineer / Researcher focused on Large Language Models to contribute to the next generation of their AI stack. This role involves developing and optimizing models for clinical documentation and structured clinical reasoning, aiming to enhance quality, cost, latency, and control. The position is research-heavy with a direct path to production, requiring the design of experiments, building datasets and evaluation systems, and fine-tuning and post-training models. The successful candidate will collaborate closely with engineering and clinical teams to deploy solutions at scale and will play a key role in shaping Knowtex's long-term LLM strategy.

Applied Intuition is seeking a Senior Machine Learning Engineer to advance the core reconstruction technology behind their Neural Simulation product. This role involves transforming real-world driving data into high-fidelity simulation environments, pushing the boundaries of reconstruction quality, scaling data processing, and enhancing product utility for customers training autonomy systems. The engineer will directly influence the development of next-generation autonomous vehicles for major OEMs, working at the intersection of 3D computer vision, graphics, and machine learning to bring cutting-edge research into production and solve complex simulation challenges in Physical AI.

$190k - $260k

We are seeking senior machine learning engineers to advance the generative modeling technology behind our Neural Simulation product. This role involves pushing the boundaries of diffusion and video generation models to create realistic, controllable sensor data and augment real-world logs with new scenarios. Your work will directly influence the development of next-generation data-driven autonomous vehicles for leading OEMs. This position is ideal for engineers passionate about the intersection of generative modeling, computer vision, and machine learning, who are eager to bring cutting-edge research into production and address complex simulation challenges in Physical AI.

$185k - $260k

Databricks

At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems by building and running the world's best data and AI infrastructure platform. As part of the AI Platform team, you will build the substrate that powers everything from data apps, AI agents, model training, model serving, and vector search. You will be joining a high-agency, high-visibility team operating at the frontier of AI infrastructure with deep ties to research, product, and real-world enterprise use cases. Databricks Mosaic AI is one of our fastest-growing businesses helping thousands of our customers democratize AI within their organizations, and we are building the infrastructure that powers the next generation of AI.

$166k - $225k

Synthesia is seeking a Staff Research Scientist to join their R&D department, focusing on the frontier of generative AI, specifically avatar-centric, interactive video diffusion models. This role involves building AI models capable of natural, real-time conversations, understanding user input (audio and video), and generating appropriate human-like reactions. The team operates in a fast-paced environment, translating cutting-edge research into tangible product features. This is a unique opportunity for experts in machine learning and diffusion models to significantly influence the future of AI video agents.

Snorkel AI is seeking a Research Scientist to bridge the gap between cutting-edge research and real-world AI systems. This hands-on role involves prototyping, building, and deploying innovative AI solutions, translating research breakthroughs into scalable, practical applications. The company operates in a fast-paced, high-impact environment, seeking individuals who thrive on rapid iteration, solving open-ended challenges, and driving innovation from research into production.

$200k - $375k

Code Metal is seeking a Staff AI Platform Engineer to architect and build the foundational AI enablement stack that internal engineering teams rely on. This role involves designing and operating production systems for GPU inference serving, model gateways, agent harnesses, context management, and observability. While primarily an engineering role focused on building and operating production systems, it also requires solid data science and AI research fundamentals to work closely with the AI Research team and conduct experiments when necessary. The platform is built to product standards, ensuring reliability and scalability for AI-driven code transpilation, modernization, optimization, modeling, and simulation solutions.

$205k - $250k

Accenture

Accenture Industry X is seeking an experienced Machine Learning Architect to join their Automotive Engineering team. This role will contribute to the development of next-generation intelligent vehicle solutions by designing and implementing Machine Learning architectures for automotive software platforms. The team focuses on shaping a future-looking Software-Defined Vehicle Platform that unifies various operating systems into a powerful developer ecosystem, driving innovation in ADAS and Infotainment.

NVIDIA is seeking a talented Machine Learning Engineer to focus on Product Security, Content Safety, and ML Fairness and Robustness for LLMs across research and production teams. This role involves assessing, quantifying, and improving the safety and inclusivity of multi-modal LLM models in a scalable manner. You will develop datasets and models for training and evaluating systems related to Content Safety, ProdSec, Robustness, and ML Fairness. The position also includes researching and implementing techniques for bias detection and mitigation in LLMs and RAG systems, defining key metrics for responsible LLM behavior, and adhering to MLOps best practices for automation, monitoring, scale, and safety. You will also contribute to the MLOps platform and develop safety tools to enhance ML team effectiveness, collaborating with cross-functional teams to address content safety and ML fairness challenges.

$124k - $242k