Scalable AI Framework Developer

3 weeks ago


Mountain View, California, United States Waymo Full time
About the Job

We are seeking a highly skilled Distributed Machine Learning Architect to join our team at Waymo. In this role, you will be responsible for designing and implementing a scalable and reliable distributed training infrastructure that can handle large-scale machine learning workloads.

Our ideal candidate will have experience with distributed systems principles, cloud computing platforms, and machine learning frameworks such as TensorFlow and PyTorch. You will have the opportunity to work on complex technical problems and contribute to the development of innovative solutions.

In this role, you will:

  • Design and implement a scalable and reliable distributed training infrastructure.
  • Develop and implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure.
  • Collaborate with cross-functional teams to develop models in Perception and Planning that are core to our autonomous driving software.

The expected salary range for this full-time position is $158,000 - $200,000 USD per year, depending on experience and qualifications.



  • Mountain View, California, United States Microsoft Full time

    The AI Frameworks team at Microsoft is responsible for developing cutting-edge software that enables the execution of AI models on a wide range of devices, from powerful supercomputers to servers, desktops, mobile phones, IoT devices, and web browsers. Collaboration with hardware teams and partners allows us to build tailored software stacks for novel AI...


  • Mountain View, California, United States Inworld AI Full time

    Inworld AI is seeking a highly skilled Artificial Intelligence Systems Engineer to join our team. This role offers an exceptional opportunity to work on cutting-edge ML projects and contribute to the development of innovative AI products.Key ResponsibilitiesDesign, develop, and implement scalable AI systems using modern frameworks and tools.Collaborate with...

  • AI Researcher

    3 weeks ago


    Mountain View, California, United States Contextual AI Full time

    Job DescriptionThe role of an AI Researcher at Contextual AI involves working on cutting-edge research projects and applying that knowledge to production systems for RAG 2.0 (Contextual Language Models + Fine Tuning + Alignment).You will be responsible for:Working on and researching state-of-the-art retrieval augmented language models as well as fine tuning...

  • AI Software Developer

    9 hours ago


    Mountain View, California, United States Microsoft Corporation Full time

    Job Summary:A highly skilled AI software developer is required to join our team at Microsoft Corporation. This role involves developing high-performance AI software for running AI models on various devices, collaborating with hardware teams and partners, and working on AI algorithmic innovation, purpose-built AI hardware, systems, and...


  • Mountain View, California, United States Microsoft Corporation Full time

    At Microsoft Corporation, we are committed to empowering individuals and organizations to achieve their full potential through technology. Our AI Frameworks team is seeking a talented Senior Software Engineer- GPU to join our ranks and contribute to the development of cutting-edge AI software.In this role, you will work closely with our hardware teams and...

  • Senior AI Developer

    3 weeks ago


    Mountain View, California, United States Bonfy Full time

    About Us: Bonfy.AI is an innovative startup working behind the scenes on groundbreaking AI innovations. Our mission is to make AI trustworthy and reliable. We're committed to innovation and excellence, and we're looking for talented professionals to join our team.Job Responsibilities:Develop, deploy, and optimize machine learning models for various...


  • Mountain View, California, United States Inworld AI Full time

    Spearhead the development of Inworld's AI Engine as our Senior AI Product Manager. Drive product vision and strategy to enable the future of AI-powered gaming and media. Collaborate closely with our AI/ML teams to determine how best to bring AI-powered solutions to game development.As a key member of our team, you will drive complex relationships with...


  • Mountain View, California, United States Inworld AI Full time

    About Inworld AIWe are Inworld AI, a pioneering company in AI-powered products, and we're looking for a highly skilled Senior Technical Product Manager to lead the development of our AI Engine.Your MissionYou will be responsible for developing the product vision and strategy of our AI engine, including API and SDKs. This requires a strong technical...


  • Mountain View, California, United States Inworld AI Full time

    Join Our Team at Inworld AIWe are looking for an experienced C++ developer to join our team in developing AI engines for games and interactive media.You will work closely with our cross-functional teams to integrate AI models into game engines, ensuring seamless user experiences.Responsibilities:Design and develop software using C++ related projects with a...


  • Mountain View, California, United States Inworld AI Full time

    Why Choose Inworld AI?Inworld AI offers a unique opportunity to work on cutting-edge AI technologies that transform the gaming industry. As a C++ Developer at Inworld AI, you will be part of a dynamic team that values innovation, collaboration, and continuous learning.Some of the key benefits of working at Inworld AI include:A competitive salary range of...


  • Mountain View, California, United States TikTok Full time

    Company OverviewTikTok is the leading destination for short-form mobile video, inspiring creativity and bringing joy to users worldwide. Our mission is to provide a platform where imaginations can thrive.Job DescriptionWe are seeking an AI Infrastructure Developer - Distributed Inference Engineer to join our team in building the next-generation AI...


  • Mountain View, California, United States Amiri Recruiting Full time

    Job DescriptionAmiri Recruiting is seeking a skilled AI Development Specialist to join our team. As an AI Development Specialist, you will be responsible for developing and maintaining our AI-powered developer tools.Company OverviewWe are a leading recruitment agency that connects top talent with innovative companies. Our mission is to empower developers to...

  • Data Quality Expert

    2 days ago


    Mountain View, California, United States Contextual AI Full time

    At Contextual AI, we are dedicated to building language models that are customizable, scalable, and trustworthy. Our mission is to create high-quality data for stakeholders across the company to evaluate and improve our AI models.The Data Team plays a crucial role in delivering high-quality data, which directly impacts the long-term success of our...


  • Mountain View, California, United States Contextual AI Full time

    Contextual AI is committed to developing high-quality language models that meet the needs of our diverse user base. To achieve this goal, we require a skilled Data Lead who can oversee the creation and management of high-quality data.This role involves defining data requirements, designing and implementing datasets, and maintaining comprehensive...


  • Mountain View, California, United States SoDA Full time

    Full-Stack Engineer Job Description:Company Overview:At SoDA, we're pioneering AI-powered video ad creation, enabling businesses to generate engaging content efficiently. With a rapidly growing user base and leadership with backgrounds from top-tier tech companies, this role offers a chance to be part of a fast-evolving environment where innovation drives...

  • AI Development Lead

    1 month ago


    Mountain View, California, United States Bonfy Full time

    Overview of Bonfy.AIBonfy.AI is a pioneering startup that focuses on making AI trustworthy. Our mission is to drive game-changing innovations and develop robust, efficient, and scalable machine learning solutions.


  • Mountain View, California, United States Microsoft Corporation Full time

    About Our TeamThe Artificial Intelligence (AI) Frameworks team at Microsoft is dedicated to developing AI software that enables running AI models everywhere. Our goal is to create a seamless experience for users, from world's fastest AI supercomputers to internet browsers.Our team collaborates with hardware teams and partners to build software stacks for AI...


  • Mountain View, California, United States Inworld AI Full time

    Inworld AI is seeking a skilled Cloud DevOps/Site Reliability Engineer to maintain and optimize our infrastructure.ResponsibilitiesInfrastructure Management: Contribute to Infrastructure-as-Code (Terraform) and maintain cloud infrastructure on AWS, Azure, or GCP platforms.Pipeline Orchestration: Develop CI/CD pipelines using Github Actions, Helm, and ArgoCD...


  • Mountain View, California, United States Tik Tok Full time

    TikTok is seeking a skilled Site Reliability Engineer to join our AML team. As a key member of our engineering team, you'll be responsible for designing, building, and maintaining our massive distributed AI/ML recommendation system. With a focus on reliability, scalability, and performance, you'll work closely with software engineering teams to ensure that...


  • Mountain View, California, United States Gatik AI Full time

    About the RoleGatik AI is seeking a highly skilled Autonomous Driving Engineer to join our team in Mountain View, CA. As a key member of our research and development team, you will play a crucial role in developing novel end-to-end algorithms for autonomy stacks, conducting research on foundation models for driving, and designing robust and explainable...