PyTorch AI Jobs

View detail

As a Research Engineer in OpenAI's Monetization Group, you will design and deploy advanced machine learning models to solve real-world problems, bringing research from concept to implementation and creating AI-driven applications with direct impact. You will collaborate closely with researchers, software engineers, and product managers to understand complex business challenges and deliver AI-powered solutions. Your work includes implementing scalable data pipelines, optimizing models for performance and accuracy to ensure they are production-ready, and monitoring and maintaining deployed models to ensure they continue delivering value. You will stay ahead of developments in machine learning and AI by engaging with the latest research, participate in code reviews, share knowledge, and lead by example to maintain high-quality engineering practices.

View detail

As a Staff Voice Research Engineer, you will lead the development of models and algorithms powering Decagon's real-time voice agents and manage multi-quarter initiatives to improve speech understanding, naturalness, turn-taking, and resilience in real-world conditions. Responsibilities include leading research and engineering efforts to enhance core conversational capabilities such as instruction following, retrieval, memory, and long-horizon task completion; building and iterating on end-to-end models and pipelines to optimize quality, efficiency, and user experience; partnering with platform and product engineers to integrate new models into production systems; breaking down ambiguous research ideas into clear, iterative milestones and roadmaps; mentoring other researchers and engineers; setting technical direction; and establishing best practices for applied research and engineering.

View detail

As a Staff Research Engineer at Decagon, you will be responsible for building industry-leading conversational AI models that power Decagon’s agent, taking them all the way from idea to production. You will own multi-quarter initiatives that enhance the agent’s reliability, capability, and efficiency. Your responsibilities include leading research and engineering efforts to improve core conversational capabilities in production such as instruction following, retrieval, memory, and long-horizon task completion. You will build and iterate on end-to-end models and pipelines optimizing for quality, efficiency, and user experience. Additionally, you will partner with platform and product engineers to integrate new models into production systems. You will break down ambiguous research ideas into clear, iterative milestones and roadmaps. You will also mentor other researchers and engineers, set technical direction, and establish best practices for applied research and engineering.

View detail

The role involves developing machine learning and artificial intelligence systems by leveraging and extending state-of-the-art methods and architectures, designing experiments, and conducting benchmarks to evaluate and improve AI performance in real-world scenarios. The candidate will participate in impactful projects and collaborate with multiple teams and backgrounds to integrate cutting-edge ML/AI into production systems. Responsibilities also include ensuring AI software is deployed to production with proper testing, quality assurance, and monitoring.

View detail

Design, develop, and deploy custom and off-the-shelf ML libraries and toolings to improve ML development, training, deployment, and on-vehicle model inference latency. Build tooling and establish development best practices to manage and upgrade foundational libraries such as Nvidia driver, PyTorch, TensorRT, to improve ML developer experience and expedite debugging efforts. Collaborate closely with cross-functional teams including applied ML research, high-performance compute, advanced hardware engineering, and data science to define requirements and align on architectural decisions. Work across multiple ML teams within Zoox, supporting in- and off-vehicle ML use cases and coordinating to meet the needs of vehicle and ML teams to reduce the time from ideation to productionization of AI innovations.

View detail

Help users discover and master the Dataiku platform through user training, office hours, demos, and ongoing consultative support. Analyse and investigate various kinds of data and machine learning applications across industries and use cases. Provide strategic input to the customer and account teams that help customers achieve success. Scope and co-develop production-level data science projects with customers. Mentor and help educate data scientists and other customer team members to aid in career development and growth.

View detail

Help users discover and master the Dataiku platform through user training, office hours, demos, and ongoing consultative support. Analyse and investigate various kinds of data and machine learning applications across industries and use cases. Provide strategic input to the customer and account teams that help our customers achieve success. Scope and co-develop production-level data science projects with our customers. Mentor and help educate data scientists and other customer team members to aid in career development and growth.

View detail

You will develop ML/AI that leverage and extend the latest state-of-the-art methods and architectures, design experiments and conduct benchmarks to evaluate and improve their performance in real-world scenarios, work on impactful projects, and collaborate with people across several teams and backgrounds to integrate cutting edge ML/AI in production systems.

View detail

The role involves translating AI research into product solutions by working with client-side researchers on post-training, evaluations, safety, and alignment, building necessary primitives, data, and tooling. The candidate will partner closely with core customers and frontier labs to address complex technical problems related to model improvement, performance, and deployment. They are responsible for shaping and proposing model improvement work by translating customer and research objectives into technically rigorous proposals, scoping work into defined statements of work and execution plans. The role requires owning the end-to-end lifecycle of projects including leading discovery, writing PRDs and technical specifications, prioritizing trade-offs, running experiments, shipping initial solutions, and scaling pilots into repeatable offerings. The candidate will lead high-stakes engagements with senior stakeholders, define success metrics, identify risks early, and drive programs to measurable outcomes. They will collaborate across teams including research, platform, operations, security, and finance to deliver reliable, production-grade solutions. Additionally, the position involves designing and implementing robust evaluation frameworks, closing feedback loops on data quality, sharing learnings, and elevating technical execution across accounts.

View detail

As a Senior Software Engineer on the Pilots team, the responsibilities include delivering robust, thoroughly tested, and maintainable C++ code for edge and robotics platforms, designing, implementing, and owning prototype perception systems that may transition into production-grade solutions, constructing and refining real-time perception pipelines including detection, tracking, and sensor fusion, adapting and integrating ML and CV models for Hayden-specific applications, driving technical decision-making balancing prototyping speed with production readiness, collaborating with the Product team and cross-functional Engineering departments, and contributing to shared infrastructure, tooling, and architectural patterns as pilots mature into foundational products.