PyTorch Jobs

Scale AI

The Machine Learning Engineer role at Scale focuses on applying advanced techniques in generative AI, computer vision, reinforcement learning, and agentic AI to enhance production environments for Scale's products and customer experience. Leveraging robust internal infrastructure and extensive datasets, ML engineers drive product improvements. The Public Sector team specifically deploys cutting-edge models for mission-critical government systems, working with large language models and computer vision. This includes developing agentic systems for complex challenges, building agent frameworks, and creating evaluation tools. Efforts also involve advancing reinforcement learning for LLMs and training advanced computer vision models for defense applications, including large-scale fine-tuning and developing generalizable foundation models.

$235k - $294k

Scale AI

Scale is seeking a Senior Machine Learning Engineer to own the application of generative AI, agentic AI, computer vision, and reinforcement learning to mission-critical production problems. In this role, you will tackle challenges without established solutions, proposing and building architectures, and ensuring their performance in customer-dependent environments. The Public Sector Machine Learning team focuses on deploying advanced models to government systems, with a primary emphasis on agentic systems built on large language models. This includes developing agents for complex operational and planning tasks, integrating retrieval pipelines and production APIs, managing state for long-running tasks, performing geospatial reasoning, and creating evaluation tooling. The role also involves applying reinforcement learning and computer vision for defense applications, including evaluation, labeling efficiency, and multimodal model training. As a Senior MLE, you will have design authority over a capability area, making final decisions on team patterns and ensuring they function under real-world constraints like classified environments, limited compute, and strict correctness requirements.

$235k - $294k

Scale AI

Scale is seeking a Staff Machine Learning Engineer for its Public Sector team to lead the design and deployment of agentic AI systems in critical government environments. This role involves working at the intersection of agentic ML, systems engineering, and applied research to build foundational infrastructure for AI systems that reason, plan, and act reliably at a national scale. You will partner directly with U.S. defense and intelligence agencies, contributing to flagship programs that advance agentic AI for geospatial reasoning, planning, and decision support. As a Staff ML Engineer, you will set technical direction, own core architectures, and translate ambitious ideas into production systems trusted by government operators.

$274k - $343k

Scale AI

Scale is seeking a Staff Machine Learning Research Engineer to join the Applied Intelligence Systems (AIS) team, which is part of the Scale Generative AI Platform. This role focuses on advancing agentic AI applications for enterprise and government use cases. You will work across the full spectrum of technical challenges within AIS, contributing to areas such as model training and fine-tuning, designing evaluation systems, building improvement loops from production data, prototyping novel agent architectures, and developing internal tooling. This is a hands-on research and engineering position where you will write code, ship production systems, and set the AI/ML technical direction for the team, influencing methods, architectures, and standards for others.

$250k - $350k

Scale AI

The Human Frontier Collective (HFC) aims to bring together top minds to shape the future of AI, focusing on the people who teach AI systems. The collective comprises researchers, academics, and domain leaders from over 70 fields, with most members holding doctorates. HFC members contribute to shaping how frontier AI systems are built, evaluated, and governed, and have co-authored significant research publications. The fellowship offers participation in high-impact AI/ML projects, opportunities to publish research, and access to an exclusive expert network.

As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X. We're looking for exceptional engineers who are passionate about our mission and have a strong desire to make a meaningful impact.

$180k - $440k

OpenAI

OpenAI is seeking a senior Machine Learning Data Scientist to lead its forecasting initiatives within the Strategic Finance team. This is a founding role for the Forecasting pillar, responsible for building and scaling robust, interpretable, and production-ready forecasting systems. The models developed will be crucial for predicting key business metrics like daily/weekly active users, revenue, lifetime value, compute consumption, and profitability, directly influencing critical business decisions and company strategy. This role requires a blend of technical expertise, product intuition, and business acumen, involving close collaboration with product managers, researchers, engineers, and finance leaders to operationalize insights and build foundational forecasting capabilities.

$347k - $445k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning Computer Architecture internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning Computer Architecture teams. Interns will work on projects with a measurable impact on NVIDIA's business, tackling complex challenges in AI and digital twins that are transforming major industries.

$0k - $0k

Databricks

Databricks is seeking a Senior Software Engineer for its AI Runtime (AIR) team. AIR is a managed platform designed for large-scale GPU training and fine-tuning of AI models, providing customers with on-demand access to accelerators and a serverless experience that simplifies the complexities of provisioning, scheduling, and orchestrating multi-node jobs. As a Senior Software Engineer on AIR, you will be instrumental in building and scaling the systems that ensure large-scale training is fast, reliable, and effortless. You will lead the architecture and evolution of the managed GPU training stack, focusing on areas such as scheduling, distributed training performance, fault tolerance, and the developer experience for large-scale job operations. Beyond hands-on system development, you will contribute to the technical vision for AIR, mentor engineers, and collaborate with product, research, and platform teams to expand the impact of custom AI training at Databricks.

$160k - $225k

Databricks

Databricks is seeking a Staff Software Engineer for its AI Runtime (AIR) team. AIR is a managed platform designed for large-scale GPU training and fine-tuning of state-of-the-art AI models, providing customers with on-demand access to accelerators and a serverless experience. As part of this role, you will be instrumental in building and scaling the systems that ensure fast, reliable, and effortless large-scale training. You will influence the architecture and development of the managed GPU training stack, focusing on areas like scheduling, distributed training performance, fault tolerance, and the overall developer experience for operating jobs at scale. This position also involves defining the long-term technical vision for AIR, mentoring senior engineers, and collaborating with product, research, and platform teams to expand the impact of custom training at Databricks.

$190k - $265k

We are seeking a visionary and hands-on Lead Machine Learning Engineer to join our Austin team. In this role, you will be the principal architect behind the next generation of our unified AI/ML platform, designing and building the infrastructure that powers everything from traditional predictive models to generative AI, large language models (LLMs), and autonomous agent frameworks. You will own the end-to-end technical strategy, blueprint, and execution of scalable backend services and data pipelines that support AI-driven applications across go-to-market, engineering, and product teams. Because our products are initiated and owned entirely within the team, you will drive the vision from initial requirements and system design to global deployment, optimization, and long-term evolutionary ownership.

Ema builds AI Employees for HR, IT, and Finance, automating busy work across the employee experience. Founded by former executives from Google, Coinbase, Flipkart, and Okta, our team comprises engineers from premier tech companies and graduates of top universities. We are backed by leading investors and are at the forefront of Agentic AI, shipping real systems that run business processes at scale. We are seeking innovative and passionate Machine Learning Engineers who excel at solving complex problems with large datasets, transforming theoretical concepts into practical, scalable solutions, and pushing the boundaries of Natural Language Processing and Information Retrieval.

$160k - $220k

Etched is building AI systems that accelerate its own development. As a Research Engineer on the RL Frontiers team, you will train proprietary frontier models to automate kernel development, ASIC engineering, production pipelines, and 3-D hardware operating systems. You will use proprietary simulators and environments to create custom data mixes and post-train models to SOTA performance on the company's hardest engineering tasks. The company is focused on inference and bets on transformer architectures, aiming to capture the entire inference market. Etched is a fully in-person team in San Jose, valuing engineering skills and expecting technical staff to contribute to both engineering and research across disciplines.

From $400k

NVIDIA is seeking a system software engineer to develop low-level diagnostic software for next-generation data center GPUs and rack-scale AI systems. The team builds software that exercises and validates complex hardware, including compute engines, memory and cache subsystems, NICs, PCIe and NVLink interfaces, power delivery, and thermal behavior. This role is ideal for an embedded, firmware, device-driver, hardware-validation, or systems software engineer who enjoys working close to hardware. You will own well-scoped components of the diagnostic software from design through implementation, validation, productization, and field support, collaborating with various engineering teams to bring up new hardware and diagnose system failures.

$124k - $196k

The Robotic Visual Perception and Learning research group at NVIDIA is seeking highly motivated PhD students for research internships. This role focuses on pushing the boundaries of computer vision, robotics, and machine learning with real-world impact. Interns will have the opportunity to publish their work in top-tier conferences and collaborate with experienced mentors, with carefully crafted projects tailored to their expertise.

$0k - $0k

NVIDIA's TensorRT team is seeking talented engineers to advance high-performance AI inference on NVIDIA GPUs. This role focuses on developing software that accelerates state-of-the-art generative AI models for local and enterprise deployments. You will tackle challenging problems in GPU code generation, just-in-time compilation, and inference optimization to deliver high performance across diverse models and hardware configurations. As a Senior Software Engineer, you will design and implement advanced GPU inference software, translating a deep understanding of GPU execution into efficient code generation and measurable improvements in generative AI inference.

$152k - $288k

NVIDIA is seeking research interns with a strong background in applying modern AI methods to re-imagine technology across Earth System Simulation and Analytics. This role involves research on applied topics such as generative data assimilation, hybrid climate simulation, full model emulation of climate model components, regional-to-global high-resolution AI autoregressive weather prediction, and dynamical downscaling. We are looking for creative, independent, and self-starting individuals to help invent the future of AI in climate science. The ideal candidate will have strong computer science skills with knowledge of physical domain sciences, or strong domain science skills with sufficient technical expertise. Join a diverse research group that tackles challenging interdisciplinary problems, publishes at top venues, and drives real impact on NVIDIA products and global climate prediction.

$0k - $0k

Abnormal AI is seeking a Senior Software Engineer to join the Detection Team, focused on building advanced technology to identify and stop email and cloud-based attacks. In this role, you will enhance the speed, responsiveness, stability, and confidence of feature development for the ML and Data Science teams. The ideal candidate possesses a first-principles approach to scalable, customer-centric solutions, a drive to solve meaningful problems, and an ownership-oriented outlook, with the ability to iterate quickly and autonomously on novel challenges.

$180k - $259k

As a new grad AI Performance Engineer, you will be responsible for bringing up new open-source foundation models on the SambaNova dataflow platform and optimizing their performance. This involves profiling model execution, identifying performance bottlenecks across compiler, runtime, and hardware layers, and applying optimizations such as quantization, kernel fusion, and improved scheduling. You will collaborate with machine learning, compiler, runtime, and hardware teams to integrate performance improvements into SambaNova products and benchmark throughput and latency to track changes across releases.

$131k - $197k

Anthropic

The Interpretability team at Anthropic is dedicated to understanding how modern language models work, believing that a deep, mechanistic understanding is crucial for making advanced AI systems safe and steerable. This role involves building and maintaining the specialized infrastructure that powers interpretability research, akin to performing 'neuroscience' on neural networks. The work spans the entire lifecycle of a production language model, from pretraining and inference to performance optimization, and directly contributes to safety audits on frontier models.

$315k - $560k