PyTorch Jobs

Fireworks is seeking new graduates for its Member of Technical Staff role, offering a unique opportunity to work on real AI systems in production from day one. As an MTS, you will write and ship code for one of the world's busiest inference platforms, serving hundreds of models at massive scale. The company matches new hires to teams based on their strengths and interests, with potential work spanning inference, model training, distributed systems, cloud infrastructure, developer platforms, or full-stack application engineering. You will be paired with a senior engineer mentor and undergo a structured ramp-up, starting with well-scoped problems and progressing to owning features and systems independently. Fireworks emphasizes a fast-paced environment with a high bar and a commitment to employee development.

$160k - $180k

Fireworks is seeking PhD graduates in Computer Science, Computer Engineering, or Electrical Engineering to join their Systems Infrastructure team. This role is for individuals who want to apply their research ideas to real-world, large-scale AI infrastructure. You will be responsible for designing and building the core systems that power Fireworks' AI platform, including schedulers, storage systems, and networks, ensuring efficient utilization of hardware and low inference latency. You will work on challenging problems related to compute placement, efficient data movement, and high-performance networking for AI workloads, with the opportunity to see your work ship in a production environment.

$200k - $230k

Blee is an AI-first marketing content compliance platform trusted by major companies like PayPal, NerdWallet, and Brex. We are a YC-backed startup defining a new category in compliance across various industries. We value in-person collaboration but offer remote opportunities for exceptional candidates, with regular off-sites and opportunities to connect across offices. You will join a team focused on clarity, ownership, and building impactful solutions.

$150k - $275k

NVIDIA AI for Media is seeking outstanding engineers to join its team, focusing on building and optimizing real-time AI solutions for media and entertainment workflows. This role will involve setting the technical direction for content authenticity AI, architecting model and pipeline strategies for detecting synthetic and manipulated media, and making critical build-vs-adapt decisions. The team works on core technologies to solve ambitious computer vision and deep learning problems, aiming to deliver studio-quality audio and high-resolution video enhancement and effects.

$272k - $431k

NVIDIA is seeking a Deep Learning Software Infrastructure Engineer to advance its Autonomous Vehicles project. In this role, you will be responsible for building and scaling the training libraries and infrastructure necessary for end-to-end autonomous driving models. By enabling training on thousands of GPUs and massive datasets, you will significantly accelerate iteration speed and enhance safety, working collaboratively with research and platform teams across NVIDIA.

$224k - $431k

Nunchux AI is building infrastructure to make multimodal generative AI faster, cheaper, and easier to build on. Founded by MIT PhDs and CMU Professor, the company leverages deep research expertise and production systems experience. The Role focuses on bringing new image and video models to the Modelverse platform, prioritizing speed to launch. You will own the initial integration, coordinate deployment with the cloud team, and work with product engineers on API and Modelverse integration. After launch, you will hand off optimization tasks and maintain the shared code for model integration and serving.

$170k - $240k

Nunchux AI is building infrastructure to make multimodal generative AI faster, cheaper, and easier to build upon. We are seeking a Machine Learning Engineer focused on post-training and evaluation to enhance the efficiency and quality of visual generative models. In this role, you will develop and validate post-training techniques, curate and manage data pipelines for training and evaluation, and build robust evaluation systems to measure model performance. You will also benchmark models against industry standards, ensure quality, and integrate cutting-edge research into production workflows.

$180k - $250k

Decagon is seeking a Research Engineer focused on Audio and Speech to build and productionize models and agent harnesses for real-time voice agents. This role involves advancing multimodal and full-duplex systems capable of natural, real-time listening, reasoning, speaking, and responding. The ideal candidate will own their work end-to-end, ship impactful improvements, and make high-level technical decisions in a collaborative environment.

$200k - $400k

Moveworks is seeking a Machine Learning Engineer to develop cutting-edge ML infrastructure for building and serving LLMs. This role is pivotal in constructing, optimizing, and scaling end-to-end machine learning systems, encompassing distributed training and inference pipelines for large language models, model evaluation and monitoring frameworks, and LLM latency optimization. The frameworks developed will serve as a robust foundation for hundreds of ML and NLP models in production, impacting millions of enterprise employees. You will address challenges in service scalability and core algorithm optimization, working closely with machine learning, data infrastructure, and core skill teams. Your contributions will significantly influence customer AI experiences and the long-term scalability of our core AI product.

Lightning AI is seeking an AI Platform Support Engineer to join their US Customer Experience team. This role is designed for individuals who are technical partners to ML teams, helping them diagnose and resolve complex issues related to large-scale training and inference workloads across cloud infrastructure, Kubernetes, and GPU platforms. You will be instrumental in supporting engineers as they train models, deploy inference systems, and scale GPU workloads in production, tackling problems ranging from Kubernetes scheduling and GPU orchestration to distributed PyTorch failures, inference latency, networking, and storage performance. This position offers exposure to diverse AI workloads and the opportunity to shape the infrastructure powering future ML applications.

$115k - $140k

Lightning AI is seeking a Senior Application Security Engineer to safeguard their AI, machine learning, and inference platforms. This is a hands-on technical role focused on integrating security into modern AI infrastructure, inference systems, and developer platforms. You will collaborate with platform engineers, ML engineers, and infrastructure teams to identify risks, design secure architectures, and develop security tooling that empowers engineers to operate efficiently and securely. The role is execution-driven, involving technical implementation, in-depth security reviews, and contributing to the growth of application security capabilities.

$180k - $220k

NVIDIA is seeking an NCX Senior Engineer to join our DSX team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. You will deliver hands-on technical assistance for advanced AI deployments, intricate distributed systems, and ensure customers realize efficient performance from NVIDIA's AI platform across varied environments. We partner with the world's most innovative AI companies to address their most challenging technical problems. In this role, you will develop innovative solutions that advance AI infrastructure capabilities and directly influence customer success with breakthrough AI initiatives.

$184k - $357k

This internship program offers a full-time, in-person opportunity in Berlin for 12 to 24 weeks. Interns will focus on advancing search quality through innovative models, data, and tools. The role involves training and optimizing large-scale deep learning models, with a particular emphasis on retrieval and ranking models. Additionally, interns will conduct research in representation learning, covering areas like contrastive learning, multilingual, evaluation, and multimodal modeling for search and retrieval. A key aspect of the role includes building and optimizing RAG pipelines for enhanced grounding and answer generation.

Physical Intelligence is developing foundation models and learning algorithms to power future physically-actuated devices. In this role, you will be instrumental in scaling and optimizing our training systems and core model code. You will take ownership of critical infrastructure for large-scale training, encompassing GPU/TPU compute management, job orchestration, and the creation of efficient, reusable JAX training pipelines. This position offers a hands-on, high-leverage opportunity at the intersection of machine learning, software engineering, and scalable infrastructure, working closely with researchers and model engineers to translate innovative ideas into production-ready training runs.

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team. This role involves working with the open-source community to enhance Apache Spark's performance and cost-efficiency on NVIDIA GPUs, enabling enterprises to scale their Spark workloads. You will apply advanced ML/AI techniques to optimize these large-scale data processing applications.

$224k - $357k

Thinking Machines is seeking a Site Reliability Engineer (SRE) to ensure the speed, reliability, and ease of iteration for our post-training and reinforcement learning (RL) systems. This role is akin to production engineering or site reliability, focused on model training infrastructure. You will be responsible for the health of training runs, clusters, and pipelines, working closely with research teams to debug issues in real-time, strengthen infrastructure, and develop tools and automation that allow researchers to focus on scientific discovery rather than job management. This position offers significant ownership, requiring you to be the point person for critical issues during active model runs and to implement solutions preventing future occurrences.

$300k - $350k

NVIDIA is seeking a Senior Software Engineer to advance the C++ foundation of CUDA Core Libraries. This role involves designing and optimizing high-performance algorithms and APIs for C++ developers, contributing to the foundational libraries, algorithms, and language/runtime infrastructure that power CUDA. You will be responsible for the full lifecycle of features, from design and implementation to testing, profiling, and maintenance, ensuring a speed-of-light experience for developers and AI coding agents.

$184k - $357k

NVIDIA is seeking a Senior Software Engineer to enhance the Python experience for its foundational CUDA Core Libraries. This role involves building Pythonic APIs, language bindings, algorithms, and runtime infrastructure that sit on top of native C/C++ foundations. You will contribute to the core libraries and infrastructure that make CUDA a high-performance platform for developers and AI coding agents. The position requires a strong focus on designing and implementing efficient, reliable, and scalable GPU-accelerated software, with an emphasis on improving the Python developer experience through robust tooling and clear documentation.

$184k - $357k

NVIDIA is seeking a motivated Deep Learning engineer to integrate advanced CUDA features and Distributed Runtime technologies into AI stacks, including frameworks like PyTorch, TRT-LLM, vLLM, SGLang, and JAX. You will join the team responsible for core CUDA features and runtimes that scale Deep Learning and HPC applications, addressing diverse multi-GPU demands from large-scale training to low-latency inference. Your work will enhance the productivity and performance of AI applications, accelerating community adoption through AI toolkits. This role offers a significant opportunity to advance the state of the art in AI technologies.

$184k - $357k