PyTorch Jobs

Blee is seeking an Applied Scientist to specialize in agent system design, focusing on architecting and implementing the next generation of intelligent compliance systems. This role involves designing frameworks where multiple agents, powered by custom and foundational models, collaborate to identify, track, and explain compliance flags. You will also be responsible for building and optimizing applied AI/ML pipelines, including model fine-tuning, prompt engineering, evaluation frameworks, and customer-specific AI configurations, and training models based on user feedback to deliver more targeted compliance recommendations.

$150k - $275k

Aleph Alpha is seeking a Senior Performance Engineer to join their pre-training efficiency team in Heidelberg. This role focuses on optimizing the systems required to train foundation models at scale, aiming to maximize hardware utilization and training throughput on large-scale GPU clusters. You will work at the intersection of deep learning frameworks, distributed systems, and GPU microarchitecture to eliminate bottlenecks from the Python layer down to the GPU kernel. The team fosters a culture of ownership, autonomy, and empowerment, with a flat organizational structure that enables quick decision-making and open communication.

Aleph Alpha is seeking a Senior AI Researcher to join our hallucination reduction capability team. This role focuses on pushing the boundaries of LLM post-training, specifically Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL), to reduce model hallucinations and improve grounding in evidence. You will be part of a team conducting cutting-edge research on foundational models, with access to hundreds of GPUs, contributing to the development of trustworthy GenAI for industrial applications.

Cohere

Cohere is seeking a Member of Technical Staff to focus on post-training large language models (LLMs) using reinforcement learning and developing methods for improved generalization. This role involves designing training recipes, environments, and evaluations to ensure models learn skills effectively and generalize beyond their training data. You will collaborate with data infrastructure, evaluation, and pre-training teams to translate research findings into robust training procedures for production models. The ideal candidate is passionate about advancing AI capabilities and contributing to the widespread adoption of AI in enterprise solutions.

CA$295k - CA$535k

NVIDIA is seeking Architects, Functional Modeling Engineers, and Simulation experts to join its Architecture Modeling group. In this role, you will collaborate with world-class architects to enable functional simulation platforms across GPU generations. You will be instrumental in modeling GPU architecture, baseboard components, and other features, working within a matrixed environment to document, design, develop tools for analysis and simulation, and validate models. The position involves familiarizing yourself with various functional and performance simulation models across NVIDIA, developing tests and testing infrastructure for new architectures, and guiding the improvement of the simulation platform to support future GPU architectures. You will also have the opportunity to develop or use AI to assist with daily tasks.

$152k - $288k

NVIDIA is seeking a Principal AI and ML Infra Software Engineer to join their Hardware Infrastructure team. This role is pivotal in enhancing researcher efficiency by implementing advancements across the entire technology stack. The engineer will collaborate closely with customers to identify and resolve infrastructure deficiencies, enabling groundbreaking AI and ML research on GPU Clusters. The goal is to build powerful, effective, and scalable solutions that shape the future of AI/ML technology.

$272k - $431k

Applied Intuition Defense is seeking a Perception Engineer to join their expert team, focusing on advancing perception model performance across various domains including maritime, land, and air-to-surface. This role involves developing custom detector and classifier architectures, leveraging proprietary MLOps tooling for rapid iteration on large datasets, and utilizing frontier AI and extensive token budgets. The position requires building and optimizing perception models for high accuracy and real-time performance on edge devices, as well as contributing to the entire perception stack from labeling tools to simulation. Collaboration with a world-class AI talent pool and scaling datasets through advanced generation techniques are key aspects of this hands-on, full-stack position.

$125k - $175k

DataRobot is seeking a Deep Learning Research Engineer Intern to contribute to the development of a multi-target joint probabilistic foundation model. This ambitious project aims to create a single model capable of handling temporal forecasting, unordered tabular regression and classification, missing-data completion, and mixed-modality inputs and outputs, learning coherent joint structures across various dimensions. The primary focus is on generating real-world value, though strong benchmark performance is also expected. This internship offers a unique opportunity to shape the architecture of a probabilistic foundation model, working on encoders, attention mechanisms, sequence architectures, and distributional output heads. The role requires a blend of deep learning expertise, strong engineering habits, and the ability to reason precisely about model capabilities and limitations.

Databricks

The Foundation Model Inference team is the backbone of Databricks’ generative AI capabilities. We build the infrastructure that enables our customers to serve, scale, and optimize frontier models with enterprise-grade reliability and performance. Our Foundation Model APIs provide a unified platform that gives customers access to LLMs with the governance, flexibility, and scalability required for enterprise production workloads. We are looking for high-agency engineers who are excited to work on powering model inference at enterprise scale.

$190k - $265k

NVIDIA Research's Human Performance and Experience (HPX) team is seeking passionate individuals for a PhD Research Intern position focused on cross-disciplinary vision science between humans and machines. This role involves collaborating with research scientists, engineers, and product engineers to address real-world problems at the intersection of computer vision, human vision science, and robotic vision. The internship offers an opportunity to contribute to AI innovation and shape the future of human-AI interaction.

$0k - $0k

NVIDIA is seeking interns for the Learning and Perception research team to work on embodied and agentic AI, including areas like perception, agentic reasoning, multi-modal and continual learning. This role offers the opportunity to contribute to research that pushes boundaries and has real-world impact, with a collaborative team environment and experienced mentors. Internship projects are tailored to match interns' expertise, and past interns have a strong track record of publishing at top venues and often return for subsequent internships.

$0k - $0k

NVIDIA is seeking a research intern to contribute to building humanoid robot foundation models and systems within the Generalist Embodied Agent Research (GEAR) group. The mission is to create general-purpose embodied agents capable of learning and mastering complex skills in both virtual and physical environments. You will collaborate with a research team known for influential work in multimodal foundation models, large-scale robot learning, game AI, and physical simulation, contributing to significant research projects and product roadmaps, including recent milestones like Project GR00T.

$0k - $0k

NVIDIA is developing world foundation models and generative AI systems for Physical AI applications, including robotics, autonomous driving, smart spaces, and embodied agents. The NVIDIA Cosmos Platform facilitates multimodal world understanding, simulation, synthetic data generation, and embodied reasoning. We are seeking talented PhD interns to contribute to advancements in Physical AI and world models. This role involves conducting research, developing algorithms, training and evaluating large-scale models, collaborating with cross-functional teams, and publishing research findings.

$0k - $0k

NVIDIA is seeking outstanding Research Interns to join the Data-Driven AI for Robotics (DAIR) group, focusing on learning embodied skills from large-scale human data. The objective is to develop AI systems that can capture, understand, and reproduce complex human motion and interaction skills across physical and digital embodiments, including humanoid robots and animated characters. This research spans the full stack, from reconstructing human motion and interactions from video to generating controllable character behaviors, transferring motion across embodiments, and training grounded controllers for robots and virtual characters. Interns will collaborate with a supportive research team producing influential work and have opportunities to connect with product teams across NVIDIA.

$0k - $0k

NVIDIA Research is seeking leading researchers in the area of simulation and AI for quantum. You should have a strong background of research excellence, software systems building experience, and interest in advancing the state-of-the-art in quantum computing and applications of GPUs to the quantum computing ecosystem. Specific areas of research interest include but are not limited to bespoke quantum representations, tensor network methods, quantum surrogate modeling, quantum inverse design, and automated proving for quantum systems. This role involves researching and developing creative and innovative algorithms and software systems for the quantum computing ecosystem, collaborating with research and product teams, and publishing original research.

$0k - $0k

NVIDIA is seeking a world-class generative AI intern to join the Fundamental Generative AI research team. This role focuses on pushing the boundaries of generative models for various data types including image, video, 3D, graph, and multimodal data, with applications in science such as drug discovery and climate modeling. The intern will collaborate within a research team known for top-tier machine learning publications and contribute to impactful, real-world advancements.

$0k - $0k

NVIDIA's Seattle Robotics Lab is seeking PhD-candidate interns to contribute to cutting-edge research in robotics and AI. This role offers a unique opportunity to work at the intersection of academic research and industry impact, focusing on the full robotics stack from perception to foundation models. Interns will play a pivotal role in developing algorithms, integrating methods into real-world systems, and contributing to high-impact publications and code releases. The lab has a strong track record of influential research and has significantly impacted NVIDIA's robotics and simulation products.

$0k - $0k

NVIDIA is seeking an exceptional computer vision and deep learning intern researcher for its spatial intelligence lab. The team focuses on research that pushes boundaries and has real-world impact. Interns will engage in research, design, and implementation of methods related to key topics such as generative AI for videos, reconstruction, simulation, video understanding, policy learning, reinforcement learning, and vision-language models. A significant outcome of the internship is the publication of original research, alongside collaboration with internal and external researchers.

$0k - $0k

We are seeking PhD research interns to advance the next generation of large language models using reinforcement learning. Our applied deep learning research team pioneers projects like Megatron, MT-NLG, and DLSS, building state-of-the-art foundation models and developing new methods to enhance their reasoning, alignment, reliability, and ability to solve real-world tasks. This internship will focus on algorithmic research at the intersection of reinforcement learning and large language models, involving the design, implementation, and evaluation of new RL-based methods for improving LLM behavior through hands-on experimentation and rapid prototyping at scale.

$0k - $0k

NVIDIA's Autonomous Systems and Physical AI Research (ASPIRE) group is seeking outstanding Research Interns to contribute to cutting-edge work in autonomous vehicles, robotics, and physical AI. This role involves tackling fundamental research challenges to enable intelligent machines to perceive, reason, and safely interact with the physical world and people. The group focuses on foundational research areas including foundation model design, embodied reasoning, safety, closed-loop training and evaluation, and agentic workflows for Physical AI. Interns will be encouraged to publish their work and open-source their code, collaborating with experts across NVIDIA to influence real-world products while conducting groundbreaking research.

$0k - $0k