PyTorch Jobs

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 strategic, ambitious, hard-working, and creative individuals for their 2027 Autonomous Vehicles and Robotics internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading teams. Interns will work on projects with a measurable business impact, contributing to NVIDIA's pioneering work in AI and digital twins that is transforming major industries.

$0k - $0k

NVIDIA is seeking ambitious and creative individuals for its Spring 2027 Developer and Performance Technology internships. These 8-12 month full-time internships offer a unique opportunity to gain hands-on experience with industry-leading Deep Learning teams, working on projects that have a measurable impact on NVIDIA's business. Interns will tackle complex challenges in areas such as AI, digital twins, gaming, robotics, self-driving cars, and virtual worlds. The program is designed for students pursuing a B.S. or M.S. degree in a relevant field, providing a chance to expand their careers and contribute to transformative technologies.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning internship program. This 12-week full-time internship offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning teams. Interns will work on projects with a measurable impact on NVIDIA's business, contributing to advancements in AI and digital twins that are transforming major industries. We are looking for students pursuing a B.S., M.S., or Ph.D. degree in a relevant field who are passionate about tackling complex challenges.

$0k - $0k

NVIDIA is seeking interns to research scalable differentiable algorithms, including adjoint and automatic differentiation methods. This role will involve connecting simulations with experimental data for calibration and data assimilation using AI/ML techniques. The work will be evaluated on simulation workloads and assessed for integration into NVIDIA software. This internship is ideal for individuals passionate about mathematics, code, hardware, and engineering 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.

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.

$195k - $345k

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.

$195k - $395k