PyTorch Jobs

Accenture

We are seeking a motivated Junior Data and AI Engineer to join our technology team. This role is ideal for a recent graduate eager to build a career in data engineering, artificial intelligence, machine learning, and cloud-based analytics. You will support the design, development, and deployment of data-driven solutions to enhance decision-making, automate processes, and create intelligent digital products. Working alongside experienced engineers and business teams, you will learn industry best practices while contributing to impactful projects.

NVIDIA is seeking a Senior Software Engineer to enhance the Rust experience for CUDA Core Libraries. This role involves designing and implementing safe, idiomatic Rust APIs for GPU computing, while integrating closely with existing C/C++ components. You will be part of the team responsible for building the foundational libraries, algorithms, and infrastructure that power CUDA, ensuring a high-performance experience for developers and AI coding agents.

$184k - $357k

Join a team that analyzes large-scale datacenter workloads on GPU-accelerated clusters. You will transform telemetry and workload data into clear findings and visuals, partnering with OS, container, GPU, and systems engineers. When useful, you will apply machine learning and deep learning techniques for categorization and forecasting, coordinating these into tools the team uses. This role involves analyzing signals to identify opportunities for application and platform improvements, working with high-dimensional data to spot trends, and communicating conclusions to engineers and leadership.

$152k - $242k

Turing is seeking Research Engineers to advance our understanding of frontier AI systems and develop practical innovations in data, algorithms, evaluation, and training. You will work at the intersection of research and engineering, investigating high-impact questions that improve real-world AI systems. You will design and run rigorous experiments, develop research-grade prototypes and tooling, and collaborate with Research, Engineering, Product, and Operations teams to translate promising ideas into scalable applications. This role offers the opportunity to contribute to both foundational research and applied AI development across synthetic and agentic data generation, reinforcement learning, post-training, model understanding, benchmarks, and evaluation.

$250k - $400k

You will help define how machine learning models run across Cloudflare’s global network, from frontier open LLMs and real-time voice models to customer-deployed models served on heterogeneous GPUs and next-generation accelerators. You’ll work with systems engineers, product teams, hardware partners, and AI/ML engineers to bring models into production with low latency, strong reliability, and efficient resource use. This role combines applied ML, inference optimization, evaluation, and production engineering, with a focus on benchmarking models, improving serving performance, validating quality, and building tooling that helps Cloudflare and its customers ship AI applications at Internet scale.

$194k - $266k

NVIDIA is seeking a Senior Software Architect to join our GPU Communications Libraries and Networking team. This role involves co-designing next-generation data center platforms and scalable communication software crucial for advancing Deep Learning and High Performance Computing (HPC). You will investigate and identify communication performance bottlenecks, design and implement new communication technologies, and explore innovative hardware and software solutions for future platforms. This is an opportunity to break performance barriers and deliver cutting-edge technologies that power AI and HPC workloads at massive scales.

$184k - $357k

NVIDIA is seeking a highly motivated senior software engineer to join its communication libraries and network software team. This role involves developing and maintaining software for complex heterogeneous computing systems that power disruptive products in High Performance Computing and Deep Learning. You will be instrumental in designing and implementing highly-optimized communication runtimes and system software that enables interactions among GPUs and with other system components, contributing to the advancement of AI and HPC.

$152k - $288k

NVIDIA is seeking a motivated Performance Engineer to join our GPU Communications Libraries and Networking team. This role involves influencing the roadmap of communication libraries that are critical for Deep Learning and High Performance Computing applications running on massive GPU clusters. You will analyze and optimize communication performance between GPUs, which directly impacts end-to-end application performance at scale. This is an excellent opportunity to advance the state of the art in GPU communication and contribute to NVIDIA's innovative technologies.

$152k - $288k

Glean is seeking a Software Engineer to help build a scalable software platform, including enterprise search solutions. This role involves working across distributed systems, data pipelines, APIs, and user interfaces to deliver secure, high-quality products. You will contribute to a platform that powers intelligent search, an AI assistant, and AI agents, with a focus on enterprise-grade AI solutions. The work is deeply embedded in how people get things done, shipping agentic capabilities on an open, extensible stack.

$187k - $234k

Glean is seeking a Software Engineer to help build a scalable software-based platform, including enterprise search solutions. You will work across distributed systems, data pipelines, APIs, and user interfaces to deliver secure, high-quality products that meet customer needs. This role involves building large-scale, fault-tolerant distributed systems and implementing data ingestion pipelines to create secure search indexes. You will also design and implement user interfaces for enterprise content search and APIs for related applications.

$187k - $234k

Faire is a technology wholesale platform connecting independent retailers with global entrepreneurs. We leverage tech, data, and machine learning to empower local businesses. The Applied AI/ML Science team builds and maintains models that power the marketplace, including shipping and delivery estimates. This role focuses on developing models that understand products from images and text to improve predictions and enhance retailer confidence, ultimately helping brands sell more on Faire.

Sieve is seeking its first Robotics Researcher to lead the training and evaluation of world-action models (WAMs) and 3D perception systems. This role focuses on understanding how data shapes robot capabilities, connecting data curation and labeling to measurable gains in robot learning. You will manage projects from inception to completion, identifying model shortcomings, determining data needs, setting up experiments, and testing results on unseen tasks and environments. A significant portion of this work involves 3D, transforming raw multimodal and egocentric video into actionable data for robot learning by incorporating geometry, pose, and action. Simulation will be used as a tool for evaluation and augmentation when beneficial, but not as the primary focus.

As a Software Engineer, Forward Deployed Engineer (FDE), you will collaborate directly with clients and internal delivery teams to develop and implement AI-powered solutions that revolutionize how work is performed. You will manage projects from inception to completion, including scoping complex problems, creating prototypes for AI workflows, and deploying robust, scalable systems built on our platform. This role involves close interaction with both technical and non-technical stakeholders, blending the roles of an AI engineer, software developer, and technical consultant. It's an ideal position for individuals eager to work hands-on with AI models and directly witness the impact of their work.

$93k - $217k

Reflection ai

Reflection is a research lab dedicated to making intelligence open and accessible for everyone. We build open models that empower individuals to control their intelligence and shape the future of AI. This role focuses on the foundational aspects of AI development, specifically in pre-training data, contributing to the creation of open foundational models. You will be part of a talented team, helping to define the company's future and the future of open AI.

Vectara is seeking a Platform Engineer to own the infrastructure that runs our deploy-anywhere platform. This role involves managing Kubernetes clusters for ML inference at scale, CI/CD pipelines, infrastructure as code, and the observability stack. You will be hands-on, writing Helm charts and Terraform, debugging issues, and shipping backend service features. The position requires deploying across AWS, GCP, and on-premises environments, including air-gapped setups, and participating in an on-call rotation to support enterprise customers.

Databricks

Databricks is seeking a Senior Staff Software Engineer to work on the AI Runtime system, which powers the world's largest AI platform. This serverless product enables companies to train their own models using enterprise data, similar to how frontier labs operate. The role involves tackling complex technical challenges to advance AI Runtime's architecture, reliability, and developer experience for both humans and AI agents. You will contribute to shaping the long-term technical direction of AI training infrastructure, anticipating industry trends, and optimizing training efficiency and cost across various model architectures and GPU accelerators.

$229k - $343k

Scale AI

The Human Frontier Collective (HFC) is a unique and exclusive expert network that brings together researchers, academics, and domain leaders from over 70 fields to shape the future of AI. With over 90% of members holding doctorates, the collective focuses on how frontier AI systems are built, evaluated, and governed. Members participate in high-impact AI/ML projects, collaborate on research publications, and gain access to a vibrant community network. This role offers opportunities to work on projects such as building AI safety guardrails, optimizing PyTorch models, and evaluating ML code, with the potential to co-author technical reports and research papers.

Scale AI

Scale's ML platform (RLXF) team builds our internal distributed framework for large language model training and inference. The platform powers MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs, as well as data quality evaluation. You will work closely across Scale’s ML teams and researchers to build the foundation platform that supports all our ML research and development, enabling the next generation of LLM training, inference, and data curation.

$190k - $237k

Scale AI

This role is at the intersection of cutting-edge AI research and practical application, with a focus on studying the data types essential for building state-of-the-art agents, such as browser and SWE agents. The ideal candidate will explore the data landscape needed to advance intelligent, adaptable AI agents, guiding the data strategy at Scale to drive innovation. This position requires not only expertise in LLM agents and planning algorithms but also creativity in addressing novel challenges related to data, interaction, and evaluation. You will contribute to impactful research publications on agents, collaborate with customer researchers, and work alongside the engineering team to translate these advancements into real-world, scalable solutions.

$302k - $378k

Scale AI

Scale's LLM post-training platform team builds an internal distributed framework for large language model training, powering MLEs, researchers, data scientists, and operators for fast and automatic training and evaluation of LLMs. This platform also serves as the underlying training framework for the data quality evaluation pipeline. You will work closely with Scale’s ML teams and researchers to build the foundation platform which supports all our ML research and development works, enabling next generation LLM training, inference, and data curation.

$290k - $363k