PyTorch Jobs

You.com is building the AI Search Infrastructure that powers modern AI systems, aiming to create a trusted knowledge layer for agents, applications, and enterprises. Our platform combines proprietary indexes with LLM-optimized retrieval systems to solve complex problems in search, large language models, and large-scale infrastructure, making AI systems more reliable, transparent, and useful. This hybrid product engineering and go-to-market role offers significant impact at the intersection of these areas, focusing on shaping our API surface and engaging directly with key customers.

$200k - $250k

We are building the AI layer of the Radio Access Network, applying agentic AI, machine learning, and LLMs to real RAN challenges. This role is for a Technical Lead to own the end-to-end architecture and technical direction of these AI systems. It is a hands-on senior individual contributor role, combining architecture, coding, technical leadership, and mentorship. You will work alongside RAN, PHY, and software teams, building production-grade AI solutions for real cellular networks.

Parallel Wireless is seeking an AI Engineering Team Manager to lead its AI team and drive its technical direction, roadmap, and execution. This role involves leading AI/ML initiatives from concept to production, defining success metrics, and ensuring AI solutions deliver measurable impact. The manager will also lead architecture and technical decisions for AI models, platforms, tools, and data, while building and growing a strong, multidisciplinary team through recruitment, mentoring, and career development. Establishing robust engineering practices for evaluation, scalability, reliability, monitoring, and model lifecycle management is key, as is close collaboration with Product Management, Systems Engineering, and R&D teams across different geographies. The role also includes partnering with domain experts to identify AI opportunities for network improvement and supporting customer-facing discussions and trials.

Ramp is seeking an Applied Scientist Intern to join their team and contribute to solving critical business problems through machine learning. As an intern, you will be a fully integrated member, owning a project from inception to completion. You will collaborate with engineers, product managers, and business stakeholders to translate complex business needs into scalable, ML-driven solutions. This role offers the opportunity to apply ML in a practical setting, ship code, and create tangible value for Ramp and its customers, focusing on areas like credit, fraud, growth, or the core product.

pika

Pika is building the next generation of AI creative tools to make video creation seamless, intuitive, and accessible to everyone. We are seeking a Principal Product Engineer to lead the technical vision, architecture, and roadmap for user-facing systems. In this high-impact role, you will guide the engineering direction for features serving millions of creative users, from intuitive UIs and collaboration tools to real-time interactions and advanced generative AI experiences. You will work closely with product, design, and engineering leadership to deliver robust, scalable, and delightful features, influencing both immediate user experience and long-term product vision from conception to launch and iteration.

NVIDIA is at the forefront of the AI revolution, exploring the next frontier of Physical AI, which involves machines that can perceive, reason about, and act within the three-dimensional world. The NVIDIA Cosmos platform provides open foundation models for Physical AI, designed to interpret images, video, and text to create a structured understanding of physical scenes, including motion, object interactions, geometry, and physical context. These models serve as the reasoning layer for robots, autonomous vehicles, and smart infrastructure. The Cosmos Engineering team is responsible for developing the core capabilities behind these models and is seeking a Senior Deep Learning Engineer to lead the data engine and end-to-end training verification for 3D spatial reasoning and perception. This role involves ensuring models accurately learn and represent 3D spatial concepts, working closely with research scientists to translate hypotheses into measurable capabilities in shipped models, and focusing on the critical role of data and evaluations in achieving frontier model quality.

NVIDIA is seeking versatile software engineers for its XLA team to build high-performance, production-grade software at the core of next-generation AI systems. This role involves developing compiler optimization algorithms for deep learning workloads, optimizing inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You will collaborate with deep learning framework and hardware architecture teams to accelerate future AI software.

$152k - $288k

Genmo is a research lab focused on developing advanced video world models to understand, simulate, and interact with the physical world, aiming to advance physical intelligence and enable robots to learn and act. We are seeking an exceptional Software Engineer to join our research team in pushing the boundaries of visual generative AI. In this role, you will collaborate with experienced researchers to develop and enhance generative models, ensuring their safe and effective deployment. This is an opportunity to contribute to cutting-edge AI research in a fast-paced startup environment.

Deepgram is seeking an Applied ML Engineer to focus on deploying their advanced speech models to a variety of edge devices and non-NVIDIA accelerators. This role involves adapting existing models to run efficiently within diverse hardware constraints, including different inference runtimes and operator sets. You will be responsible for optimizing models through techniques like quantization and precision adjustments, validating performance on real hardware, and building robust deployment pipelines for edge targets. This is a hands-on engineering position, distinct from research or cloud-focused roles, aimed at ensuring Deepgram's cutting-edge voice AI can be deployed effectively across a wide range of customer hardware.

$155k - $245k

LanceDB is seeking an AI Research Engineer to collaborate with their research team on open-ended research projects. The role focuses on end-to-end training flows across various AI domains, demonstrating how LanceDB can accelerate research. This is a unique opportunity to pursue research interests as an engineer, significantly impacting product development and raising awareness of LanceDB's capabilities.

$150k - $255k

Plaid is seeking a Senior Machine Learning Engineer to join their Data team within the Fraud organization. This role will focus on building and scaling machine learning systems that power Plaid's fraud detection products, leveraging unique network data to identify and stop fraud. The team owns the full ML lifecycle, from feature pipelines and model training to production serving and monitoring, building reliable and scalable systems. As a Senior Machine Learning Engineer, you will own the development of high-performance feature computation and online inference pipelines, build robust observability and monitoring capabilities, and partner closely with ML Infrastructure, Data Science, and Product teams to deliver high-impact ML solutions.

$229k - $315k

Plaid is seeking a Senior Machine Learning Engineer to join their Fraud Data team. This role will focus on developing and deploying machine learning systems to enhance Plaid's fraud detection products. You will work across the full data science and machine learning lifecycle, from identifying new signals and experimenting with models to deploying and optimizing them in production. The goal is to continuously learn from real-world performance and customer feedback to protect customers from evolving fraud threats. You will lead projects from initial experiments through model deployment and ongoing improvement, identifying predictive patterns in Plaid's network data.

$229k - $315k

Ambient.ai is seeking an AI Infrastructure Software Engineer to design, build, and optimize the AI infrastructure powering their real-time intelligence platform. This role involves working on systems to run state-of-the-art deep learning models, including computer vision, large language models (LLMs), large vision models (LVMs), and multimodal AI systems, across vast amounts of video data in real time. You will focus on scaling infrastructure for inference, evaluation, and continuous model improvement, partnering closely with research scientists and product engineering teams to bring AI advancements into production.

$168k - $205k

Mintlify is seeking product-minded engineers to build the AI experiences behind their documentation platform. This role involves combining strong full-stack engineering with applied AI expertise to create agents that reliably solve real customer problems. As a Senior Applied AI Engineer, you will take ownership of complex AI products from early experimentation through production, helping to define technical direction, establish evaluation methods for agent performance, and elevate the quality of both AI products and engineering practices. You will architect and ship customer-facing AI experiences, build reliable agent systems, drive evaluation and iteration, and transform prototypes into scalable, reliable products.

$190k - $265k

NVIDIA is seeking a Senior Software Engineer to specialize in Deep Learning Inference. This role involves designing, building, and optimizing GPU-accelerated software for AI applications, focusing on high-performance deep learning frameworks like SGLang and vLLM for efficient large-scale model serving and inference. You will contribute to improving these platforms, enabling the smooth deployment of advanced language models and working with the deep learning community to implement and release the latest algorithms. The work will concentrate on enhancing the performance of state-of-the-art LLM and Generative AI models across various NVIDIA accelerators, utilizing open-source tools and kernels for optimization.

$184k - $357k

NVIDIA is seeking a highly technical Engagement Lead to drive the integration of Generative and Agentic AI software with enterprise ISV partners. In this role, you will serve as a trusted advisor to Software Platform Architects and unite teams behind a unified technical vision. You will remain hands-on, designing and shipping production agentic systems for demanding workloads while accelerating the adoption of NVIDIA's stack across key business initiatives. This position requires versatility at the intersection of platform strategy, a strong AI/ML background with hands-on execution, and the communication skills to engage with senior leaders. It's a high-impact role offering broad skill development and exposure to the cutting edge of Agentic AI, with insights directly shaping NVIDIA's product development.

$224k - $431k

Parallel Wireless is seeking a skilled wireless systems engineer to lead the development of a multi-RAT digital twin for their Open RAN solution. This role involves executing production RAN software within a closed-loop simulation environment, enabling engineering teams to design, evaluate, and compare features for LTE, 5G NR, and 2G without the need for physical radio setups. The position is a senior individual contributor role at the intersection of wireless systems, simulation, production software, and AI/ML, focused on evolving an existing simulator into a scalable platform for feature development, testing, and optimization.

Anthropic

The Safeguards ML Inference Path team designs, builds, and operates the production infrastructure that powers Claude's ML-based safety systems. This role involves collaborating closely with safety researchers and inference engineers to bring new classifiers and novel classes of ML defenses to production, owning the research-to-production transfer of new safety technologies critical for every Claude model launch. The team builds for scale, serving thousands of ML classifiers for all requests on the token generation path and for every platform Claude runs on. Engineers will work at the intersection of machine learning, large-scale distributed systems, and AI safety, developing platforms and tools that enable safeguards to operate reliably at scale, with tooling and infrastructure used for every model launch.

$320k - $485k

Parallel Wireless is seeking experienced wireless algorithm experts to research and design advanced cellular communication algorithms for 5G and beyond products. This role will leverage neural networks and machine learning techniques to reimagine mobile networks with innovative, energy-efficient Open RAN solutions. The ideal candidate will contribute to the future of telecommunications by driving innovation through green and sustainable networks.

At Suki, we are building technology that automates clinical documentation in real-time using cutting-edge AI, significantly reducing the administrative burden on clinicians. Our platform powers voice-enabled healthcare experiences across various platforms, aiming to make clinicians' lives better. The Machine Learning team is central to this mission, driving innovation and enhancing our AI capabilities. You will leverage your NLP expertise to improve our computational infrastructure, technical approaches, and algorithms. Working directly on the platform, you'll collaborate with cross-functional teams to tackle NLP challenges and scale our services. This role involves managing and curating large datasets, monitoring system performance, and staying abreast of the latest advancements in NLP and Generative AI to maintain our competitive edge.

$225k - $250k