PyTorch Jobs

OpenAI

The Monetization team is a new cross-functional group focused on building foundational systems to responsibly scale access to intelligence. Our mission is to develop user-first, privacy-preserving monetization products, including next-generation ads experiences, that strengthen user trust, unlock economic opportunity, and support OpenAI's long-term innovation. This role operates in a greenfield environment, moving quickly through prototyping, experimentation, and iterative deployment, partnering closely with Product, Design, and Research to bring research breakthroughs into real-world systems at global scale. As a Research Engineer, you will contribute to deploying state-of-the-art models in production environments, helping turn research breakthroughs into tangible solutions and making AI technology accessible and impactful.

$381k - $555k

Applied Intuition is building a robot learning platform to enable any robot to learn industrial tasks and continuously improve. The robotics team develops this platform and uses it to deliver robot autonomy on customer lines, training, evaluating, and deploying policies on real robots performing industrial tasks. This role involves hands-on work with hardware, seeing your contributions reach customers, and working on manipulation policies from task definition to reliable execution on a physical robot. You will train and fine-tune policies, diagnose failures, and ensure their dependability for customer use, with success measured on the robot itself.

$150k - $300k

Databricks

Databricks AI is seeking a Principal AI Research Scientist to lead a team focused on advancing the state of the art in large-scale machine learning, particularly in LLM training and inference efficiency. This role involves defining and executing a research roadmap to improve how customers train, serve, and adapt LLMs at scale, working closely with product, data, and engineering leaders. The ideal candidate will drive algorithmic innovations, optimize ML systems for distributed training and inference, and translate research breakthroughs into customer-impacting capabilities within the Databricks AI platform. This position also requires fostering a culture of scientific excellence, representing Databricks AI externally, and mentoring research talent.

$270k - $340k

Databricks

Databricks AI Research is seeking a Principal Research Scientist – Scaling to lead a team focused on advancing the state of the art in large-scale machine learning, particularly in LLM training and inference efficiency. This role involves defining and executing a research roadmap to improve how customers train, serve, and adapt LLMs at scale, working closely with product, data, and engineering leaders. The goal is to push the boundaries of LLM efficiency beyond existing models, exploring novel algorithms, systems, and infrastructure improvements. You will drive independent research agendas and collaborate on low-level implementation details to bring cutting-edge methods into production and enhance the Databricks AI platform.

$280k - $350k

Databricks

The Applied AI team at Databricks is at the forefront of advancing GenAI-powered products, having launched Databricks Assistant, AI/BI Genie, and Agent Bricks. These products are used by hundreds of thousands of users daily, tackling challenging problems like code suggestion, text-to-sql generation, and knowledge QA. As GenAI products evolve, we are seeking engineers to drive the next phase of development, focusing on enhancing LLM quality, expanding GenAI capabilities, and strengthening platform architecture for AI interactions at scale.

$190k - $285k

Databricks

At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems. We build and run the world's best data and AI infrastructure platform. As part of the AI Platform team, you'll build the substrate that powers everything from data apps, AI agents, model training, model serving, and vector search. You'll be joining a high-agency, high-visibility team operating at the frontier of AI infrastructure with deep ties to research, product, and real-world enterprise use cases. Databricks Mosaic AI is one of our fastest-growing businesses helping thousands of our customers democratize AI within their organizations. We’re building the infrastructure that powers the next generation of AI.

$165k - $220k

Databricks

Databricks is seeking a Staff Backline Engineer to join their critical bridge team between Frontline Support and Engineering. This role focuses on handling complex technical issues and escalations within the Data and AI ecosystem, with a strong emphasis on customer success and proactive issue resolution. The engineer will be a technical expert and escalation point for challenging ML/AI issues, working collaboratively with Support, Engineering, Product, and customers to troubleshoot, identify root causes, and drive resolutions. The team prioritizes automation and tooling to enhance troubleshooting efficiency and improve platform supportability.

$0k - $0k

Databricks

Databricks is seeking a Staff Technical Lead for Customer Experience Intelligence (CXI) to architect and deliver an AI-native vertical workflow platform. This platform will orchestrate AI agents, humans, and enterprise tools into end-to-end operational workflows, initially focusing on customer support and expanding to other domains like IT, Security, Legal, and Marketing. The role involves driving rapid backend development with a focus on quality, safety, and reliability, designing state-machine and agentic workflows, and implementing robust observability and rollback mechanisms for AI-driven decisions. The Technical Lead will also act as a liaison between CXI, Product, Platform Engineering, and business stakeholders, while mentoring engineers and subject matter experts.

$190k - $261k

Databricks

Databricks is seeking a Staff Software Engineer to join their AI Research Infrastructure team. In this role, you will be responsible for developing and operating the research stack that powers Databricks' AI research efforts. You will design and build services for scheduling, orchestrating, and observing large-scale training and inference experiment workloads across thousands of GPUs. Your work will involve improving developer tooling to ensure researchers can iterate quickly while maintaining reliability, efficiency, and security. You will collaborate closely with research scientists, ML engineers, and platform teams to transform experimental workloads into robust, repeatable pipelines and push the boundaries of the infrastructure's capabilities.

$199k - $270k

Databricks

Databricks is seeking an engineer to help lead the development of the data and agent layer for enterprise digital advertising. This role involves building solutions to address the significant data and intelligence gaps in advertising campaign performance, which currently relies on slow, manual, and often inaccurate processes. You will be instrumental in creating systems that ingest, normalize, aggregate, and analyze advertising data at scale, enabling better decision-making and optimization for enterprises spending billions on digital ads.

$191k - $254k

We are seeking a mission-driven Developer Relations Manager focused on Foundational AI Research to engage leading academic labs advancing the next generation of AI models, systems, and methods. In this role, you will work directly with top researchers building frontier AI systems, including large language models, multimodal models, reasoning systems, training methods, inference systems, model serving, and scalable AI infrastructure. You will help researchers adopt NVIDIA’s AI and accelerated computing platforms to push the boundaries of model performance, efficiency, and scale.

$152k - $288k

We are seeking a Senior Machine Learning Engineer to join our Clinical Team. In this role, you will be responsible for owning the tissue-class segmentation and labeling models for our ultrasound CT clinical analysis layer, along with the pipelines that support their training and verification. You will retune models as image inputs are updated and clinical indications expand, and define training/evaluation pipelines, datasets, and metrics from the ground up. A key part of this role involves collaborating with data labeling contractors, expert clinicians, and internal teams on labeling specifications, quality control, and dataset versioning. You will also play a crucial part in productionizing models into a versioned, HIPAA-bound analysis service, ensuring reproducible inference, per-prediction confidence, drift monitoring, and safe fallbacks.

$237k - $250k

We are seeking a skilled engineer to join our Autonomous Vehicles team, focusing on the L3/L4 autonomous-driving solution. You will be instrumental in developing a comprehensive, learned 3D/4D world model by fusing navigation, ego-motion, perception, and sensor signals. This role involves close collaboration with perception, prediction, planning, and simulation teams to create a world representation that is complete, temporally consistent, uncertainty-aware, and robust for challenging driving scenarios. Our vision is to develop a shared multimodal scene representation that supports road understanding, 3D object perception, motion prediction, and planning, aiming to achieve human-level AI for navigation in unconstrained environments.

NVIDIA is seeking a Senior Integration Engineer to join their autonomous driving team. This role focuses on accelerating the development, integration, evaluation, and deployment of end-to-end driving models. You will collaborate across model development, data, simulation, systems software, and vehicle engineering teams to transform evolving AI models into reliable autonomous driving functionality on NVIDIA's computing platforms. The position involves working with large-scale training infrastructure, simulation environments, and production vehicle platforms.

$152k - $288k

NVIDIA is seeking a Senior Perception Software Engineer to design, develop, and productize autonomous driving solutions. As a member of the perception team, you will build world-class perception solutions, designing and implementing software components. You will be challenged to improve the robustness, efficiency, and reusability of these components, working closely with experienced perception engineers and researchers. This role involves executing the full software development lifecycle, writing well-designed, testable code, and integrating perception software into a functional system. You will also develop verification plans, document and maintain software functionality, and debug existing systems, ensuring alignment with project plans and industry standards. A key aspect of the role is deploying state-of-the-art perception DNN models to embedded platforms and analyzing system performance using benchmark data.

$224k - $357k

At Thinking Machines, we build tools that empower individuals to customize AI models for their specific needs, including the ability to train model weights. In this role, you will focus on frontier customization techniques and contribute to developing a leading post-training engine called Tinker. Your work will involve a comprehensive understanding of Reinforcement Learning (RL) science, and your findings will directly influence Tinker's training defaults, API design, and the open-source Tinker Cookbook. You will collaborate with internal research teams and contribute to open science initiatives for external partners.

$350k - $475k

Thinking Machines is building AI that extends human will and judgment, with a focus on training frontier models and developing tools that allow users to customize AI for their unique needs. This role leads the Fine-tuning Science team, focusing on frontier customization techniques and the Tinker post-training engine. It's a player-coach position where you will remain hands-on with scientific research while also guiding the team, shaping its direction, and ensuring research findings are integrated into Tinker. You will collaborate with internal research teams, contribute to open science, and engage with external users.

$350k - $475k

Gecko Robotics is seeking an AI/ML Engineering Intern to join their Engineering team for Summer 2027. This is a full-time, onsite internship where you will work on a defined project that turns inspection data into decisions, building models and the systems around them that run reliably on real customer data. This role focuses on building the systems that analyze data, rather than just the analysis itself. Candidates must be authorized to work in the United States due to government program requirements.

SpaceXAI is building Super Intelligence to help humanity understand the universe, from supercomputers to Grok models and their suite of applications. The RL Environments team is seeking a Software Engineer to build systems that transform real-world data into high-quality environments and tasks for reinforcement learning. This role involves creating more realistic training environments and owning the platforms and quality layers for trustworthy data production at scale. You will collaborate closely with ML Platform and research teams to convert various data sources into training-ready environments, making high-quality RL data more accessible for model training efforts.

NVIDIA is seeking a Principal Systems Software Engineer for Semiconductor Inspection to join their team in Santa Clara. This role involves defining, building, and evaluating AI systems for semiconductor manufacturing, focusing on inspection and metrology. The work spans computer vision, time-series modeling, multimodal AI, and anomaly detection, with a strong emphasis on adapting and deploying these technologies within partner environments. The goal is to create integrated, reliable, and fast AI solutions that can be operated by fabs and equipment vendors, even in air-gapped facilities with limited data and tight production budgets.

$272k - $431k