PyTorch Jobs

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

NVIDIA is developing advanced 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 the advancement of Physical AI and world models.

$0k - $0k

NVIDIA Research is seeking PhD interns for the AI-Aided Engineering (AIE) team. This role involves developing novel AI methods for physical simulation and engineering design, with a focus on neural operators, geometric representations, and large-scale learning. Interns will work with experienced mentors on impactful research, with opportunities to publish in leading conferences and journals. Projects will be tailored to interns' expertise, aiming to advance the field and transfer research into practical applications.

$0k - $0k

NVIDIA is seeking senior engineers to enhance its AI platform, focusing on performance optimizations within deep learning frameworks like JAX. The goal is to create a robust, efficient, and scalable platform for data handling, training, and analysis across various deep learning solutions. This role requires strong programming and system design skills, along with excellent communication and planning abilities, to contribute to the next generation of numerical computing and machine learning research tools.

$184k - $357k

NVIDIA is seeking an extraordinary Senior Perception Engineer to develop and productize autonomous driving solutions. This role focuses on creating end-to-end perception modules responsible for understanding traffic signals, signs, road markings, and construction objects, crucial for driving without reliance on HD maps. The goal is to enhance the robustness, accuracy, and efficiency of these solutions to enable autonomous driving in all conditions. You will be instrumental in designing perception systems, applying deep learning for various traffic-related tasks, developing generalizable approaches for diverse driving scenarios, and driving data-driven development by collaborating with data collection and labeling teams. Leveraging data simulation and augmentation will be key to solving extreme scenarios, and you will be responsible for productizing these solutions to meet stringent safety, latency, and software robustness requirements.

$184k - $357k

PermitFlow is seeking an Applied Machine Learning Engineer to develop the ML foundation for their AI agents, which are redefining how America builds by tackling complex information challenges in the construction industry. This role involves designing, prototyping, and deploying intelligent systems for document processing, insight extraction, and autonomous permitting workflows. You will own the end-to-end ML lifecycle, from research and data engineering to production deployment and continuous evaluation, contributing to the AI infrastructure that accelerates housing, clean-energy, and infrastructure development.

$175k - $250k

Aleph Alpha is seeking a Senior AI Researcher to join their Pre-training team, focusing on advancing the architecture and training of next-generation foundation models. This role involves designing inference-efficient architectures, optimizing large-scale training recipes, and working with thousands of NVIDIA GPUs. The position is ideal for someone excited about the intersection of research and engineering, translating mathematical reasoning into principled training decisions from small-scale experiments to large-scale runs. The researcher will influence model quality, run reliability, and efficiency by combining rigorous experimental design with high-quality production code.

At voize, we are building AI companions to seamlessly take over digital workflows for frontline workers, giving them more time to care, connect, and be present. Our mission is to amplify human impact through technology, not replace it. Backed by significant funding and trusted by thousands of facilities and hundreds of thousands of users, we combine cutting-edge technology with meaningful social impact. We are looking for a Senior ML Engineer to join our dynamic team and contribute to our mission.

We are an applied AI lab building end-to-end software agents, known for creating Devin, the first AI software engineer. Our team comprises world-class talent from leading AI companies. We are seeking a founding member for our safety team to tackle critical challenges in AI safety and alignment. This role is hands-on, focusing on ensuring our AI agents behave correctly and safely in real-world applications, rather than solely in research benchmarks. You will play a pivotal role in shaping the safety and alignment of our AI systems.

Code Metal is seeking a Senior Software Engineer for its AI Platform team. This role involves designing, building, and operating the foundational AI enablement stack that powers the company's AI-driven code solutions. You will contribute to critical components such as GPU inference serving, model gateways, agent harnesses, context engineering, observability, and AI experimentation management. While this is an internal platform, it is built to product standards. The position offers the opportunity to eventually own a specific area, like model serving or agentic infrastructure, while also contributing broadly across the stack. The role requires a blend of strong software engineering skills with solid data science and AI research fundamentals, including occasional experimental work to inform platform decisions.

$170k - $210k

NVIDIA is seeking a Deep Learning Compiler Engineer to join its Deep Learning Compiler (DLC) team. This role involves developing and optimizing compiler algorithms for deep learning networks, which are crucial for NVIDIA's inference engine across various platforms. The compiler is designed to deliver leading inference performance, fast build times, reduced memory footprints, and ease of use for the global deep learning community. You will collaborate with deep learning software framework and hardware architecture teams to accelerate the next generation of deep learning software, contributing to defining public APIs, performance analysis, and implementing compiler infrastructure techniques.

$152k - $242k

NVIDIA is seeking a Senior Deep Learning Software Engineer to join their Algorithmic Model Optimization Team. This role focuses on optimizing generative AI models, including LLMs and diffusion models, for maximum inference efficiency. The team utilizes techniques like neural architecture search, pruning, sparsity, and quantization, and develops an innovative software platform for automated deployment. The engineer will contribute to applied research and the development of this software, which is used both internally at NVIDIA and externally by other AI development teams. This is an opportunity to push the boundaries of inference efficiency and large-scale automated deployment, working with high-level frameworks and improving high-performance kernel implementations.

$184k - $357k

Handshake is seeking a Machine Learning Engineer I for the Growth Relevance team to help shape the evolution of careers in the AI economy. This role will focus on developing, deploying, and enhancing machine learning systems for lifecycle optimization, personalized notifications, and monetization strategies. You will work with cutting-edge ML infrastructure, including embedding-based retrieval, Graph Neural Networks, and multi-stage rankers, contributing to critical marketplace metrics and responsible AI practices.

$150k - $189k

Luma is seeking a Research Scientist/Engineer to build the systems that enable reinforcement learning (RL) at a frontier scale. This role involves coupling policy optimization with large fleets of inference workers, agentic environments, and reward/verification systems to translate model behavior into learning signals. RL is crucial for Luma's models to evolve from capable to useful. This is a full-loop systems problem encompassing training, rollout generation, environment execution, and reward computation across thousands of GPUs, requiring speed, stability, and correctness. The ideal candidate has experience with post-training LLMs using RL, building environments and verifiers, and debugging large-scale asynchronous rollout pipelines.

Luma is seeking a Software Engineer to own the serving of their AI models. This role involves integrating new model architectures into the inference engine, scaling deployments across thousands of machines, and optimizing the utilization of expensive GPU fleets while meeting internal service level objectives (SLOs). The work focuses on large-scale inference systems, including scheduling, fleet management, deployment pipelines, and ensuring reliability across diverse clusters and hardware providers. This position is ideal for a strong systems engineer comfortable with model serving and Kubernetes at scale, rather than those seeking pure modeling work.