Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Principal PM: Frontier Model Training on Trainium

Socket.dev

Amazon Annapurna Labs in Seattle is hiring a Principal Technical Product Manager to define and drive the training software strategy for Trainium. You will shape distributed training libraries, post-training workflows, and performance optimization to empower researchers and operators to train frontier models at scale. You will partner with engineering, applied science, and PM teams to deliver reusable building blocks, drive adoption, and lead end-to-end launches across the training ecosystem on #J-18808-Ljbffr Socket.dev

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Principal PM: Frontier Model Training on Trainium in Seattle, WA vacancy
  •  ...responsible for building machine learning models and systems to protect our users from...  ...reached its limit. This topic focuses on two frontier directions: (1) Multimodal moderation...  ...: We study large-scale MoE architecture training and routing optimization, cross-modal... 
    Training
    Flexible hours
    Shift work

    TikTok

    Seattle, WA
    2 days ago
  • $182.8k - $247.3k

    As a Principal Solutions Architect supporting Foundation Model Providers (FMP) on AWS, you will tackle some of the most challenging and exciting problems in cloud...  ...cloud architectures that enable model providers to train, fine-tune, and serve state-of-the-art generative... 
    Principal
    Training
    Flexible hours

    AmazonWebServices

    Seattle, WA
    22 hours ago
  • $177k - $239.4k

    Amazon's Frontier AI organization is building the next generation of foundation models that power experiences across the company. We are looking for a Principal Technical Program Manager to own the end-to-end...  ..., access, and govern the training assets used to develop AI models... 
    Principal
    Training
    Flexible hours

    Amazon

    Bellevue, WA
    22 hours ago
  • $57 per hour

    Responsibilities Work on building machine learning models and systems to protect user content as part of the Trust & Safety team. Contribute...  ...of multimodal moderation foundation models, focusing on training stability, routing optimization, cross‑modal alignment, and... 
    Training
    Hourly pay
    Internship
    Local area
    Flexible hours

    TikTok

    Seattle, WA
    22 hours ago
  • $305k

     ...Product Manager on Claude Code's model performance team, you will...  ...our models. We're looking for a PM who has personally built...  ...the connective tissue between frontier research and the millions of developers...  ...combination of education, training, and/or experience Required field... 
    Training
    Work at office
    Visa sponsorship
    Flexible hours

    United States Digital Space LLC

    Seattle, WA
    3 days ago
  • $75k - $100k

     ...Monday through Friday from 8:00 AM to 5:00 PM, with occasional evening shifts as...  ...n Experience with telehealth delivery models and familiarity with virtual counseling tools...  ...care team, you will have access to ongoing training, mentorship programs, and opportunities... 
    Training
    Full time
    Work at office
    Immediate start
    Remote work
    Monday to Friday
    Afternoon shift

    Radiance Counseling Center LTD

    Bellevue, WA
    21 days ago
  • $181.1k - $245k

     ...earliest innings. As AI training, inference, and...  ...problems get solved.As a Principal PMT, you will own product...  ...operating at the frontier of performance. The AI...  ...products built for AI model training and inference...  ...utilization across GPU, Trainium, and Inferentia deployments... 
    Principal
    Training
    Flexible hours
    Shift work

    Amazon

    Seattle, WA
    2 days ago
  • $224k

     ...technical, builder-mindset Principal Product Manager to own...  ...a builder. The Principal PM will own the channel-level...  ...journey: from how LLMs are trained and source data, to how...  ...Thesis BuildingMonitor the frontier AI ecosystem continuously: model releases, product updates... 
    Principal
    Training
    Full time
    Shift work

    Expedia

    Seattle, WA
    22 hours ago
  • $180k

     ...As a multimodal engineer on the Imagine Model Team, you will develop cutting-edge AI experiences...  ...span data curation, modeling, training, inference serving, and product integration...  ...with product teams to push model frontiers and deliver exceptional end-to-end user... 
    Training
    Temporary work

    SpaceXAI

    Seattle, WA
    5 days ago
  • $202.16k - $368.22k

     ...Infrastructures team oversees the distributed training, reinforcement learning framework, high-...  ...technologies for AI foundation models. Responsibilities Design and build scalable...  ...intelligence. We aspire to advance the frontier of intelligence to drive progress for... 
    Training
    Temporary work
    Internship
    Local area

    ByteDance

    Seattle, WA
    22 hours ago
  • $35 - $100 per hour

     ...AI data business in history. We work directly with frontier AI lab researchers to create evaluations, publish...  ...AI advancement. Frontier AI labs currently improve model capabilities with various data-intensive post-training techniques. We believe that data spend for AI... 
    Training

    Handshake

    Seattle, WA
    3 days ago
  •  ...AI data business in history. We work directly with frontier AI lab researchers to create evaluations, publish...  ...AI advancement. Frontier AI labs currently improve model capabilities with various data-intensive post-training techniques. We believe that data spend for AI... 
    Training
    Immediate start
    Relocation

    Handshake

    Seattle, WA
    3 days ago
  • $35 - $120 per hour

     ...AI data business in history. We work directly with frontier AI lab researchers to create evaluations, publish...  ...AI advancement. Frontier AI labs currently improve model capabilities with various data-intensive post-training techniques. We believe that data spend for AI... 
    Training

    Handshake

    Seattle, WA
    3 days ago
  • $240k - $379.5k

     ...to growFacilitating feedback loops between our platform and model builders to continuously improve performance, usability, and...  ...stand out from the crowd:Product manager experience at a company training frontier AI models and building AI infrastructureHardworking and... 
    Principal
    Training
    Full time
    Work experience placement

    Nvidia

    Seattle, WA
    22 hours ago
  •  ...plan, adapt, and act autonomously.As a Principal Applied Scientist, you'll help invent...  ...lead the research and development of frontier AI systems that combine foundation models, multimodal reasoning, agentic workflows, and post-training techniques to build autonomous... 
    Principal
    Training
    Work at office
    Immediate start
    Remote work

    UiPath

    Bellevue, WA
    4 days ago
  • $272k - $431.25k

     ...FlashDreams are the foundation for a new generation of interactive world-model systems. With this release, NVIDIA has set the standard for...  ...causal video generation, self-forcing or causal-forcing style training, Gaussian splatting, NeRFs, and neural reconstruction.... 
    Training
    Full time

    Nvidia

    Seattle, WA
    1 day ago
  •  ...Tech Lead / Principal Engineer, Creator Agent Algorithm InfrastructureResponsibilitiesThe...  ...: Provide efficient training, evaluation, and iteration...  ...RL, SFT, and additional frontier optimization directions (see...  ...tree search, process reward model–guided reasoning, etc.);Tool... 
    Principal
    Training
    Temporary work
    Shift work

    ByteDance

    Seattle, WA
    1 day ago
  • $198.9k - $269k

    The Principal Applied Scientist will own the science mission for building...  ...language processing, LLM training and application, and agentic...  ...systems to advance the scientific frontiers of autonomous intelligence...  ...through advanced predictive modeling and contextual understanding;... 
    Principal
    Training
    Worldwide
    Flexible hours

    Amazon

    Bellevue, WA
    22 hours ago
  • $203.5k - $275.3k

     ...software that powers the Inferentia and Trainium-based Inf1, Inf2, Trn1 an Trn1n families...  ...providing the highest-scale, distributed Training and Inference servers available. We...  ...Machine Learning services.We're seeking a Principal Program Manager for the Annapurna ML, Neuron... 
    Principal
    Training
    Local area
    Remote work
    Flexible hours

    Amazon

    Seattle, WA
    3 days ago
  • $198.9k - $269k

     ...We are seeking an exceptional Principal Applied Scientist to drive...  ...visual reasoning foundation models. You will be a technical leader...  ...reasoning engine powered by frontier Large Video Models (LVMs). Your...  ...computational challenges to train foundation models at scale, taking... 
    Principal
    Training
    Worldwide
    Flexible hours

    Amazon

    Seattle, WA
    22 hours ago
  • $179.9k - $243.4k

     ...is looking for an experienced Principal Product Manager to own the vision...  ...scaleThis role is ideal for a PM with deep experience in the...  ...systems and ML/science-based models.* Collaborate with applied scientists...  ..., including architecture, training/inference lifecycles, and... 
    Principal
    Training
    Flexible hours

    Amazon

    Seattle, WA
    1 day ago
  •  ...Principal Product ManagerSeattle, WAWho We AreEach of us at Yoodli has faced our own communication...  ..., SAP, FranklinCovey, KornFerry, Sandler Training, Toastmasters, and many more — with...  ...at yoodli.ai.About The RoleWe're hiring a PM who can operate with high ownership in a... 
    Principal
    Training
    Work at office
    3 days per week

    Yoodli AI Roleplays

    Seattle, WA
    1 day ago
  • $182.8k - $247.3k

     ...run their networks. You will bridge the gap between frontier AI capabilities (large language models, multimodal foundation models, retrieval-augmented generation...  ...non-technical audiences and at C-level, including training, workshops, publications- Experience developing... 
    Principal
    Training
    Flexible hours
    Shift work

    AmazonWebServices

    Bellevue, WA
    2 days ago
  • $181.1k - $245k

     ...tailored to their workflows, mental models, and goals. We are a fast-...  ...THE ROLEModel a "builder PM" culture where hands-on experimentation...  ...exception.We are seeking a Principal Product Manager - Technical to...  ..., Well-Architected, Training & Certification) to ensure persona... 
    Principal
    Training
    Immediate start
    Flexible hours

    Amazon

    Seattle, WA
    22 hours ago
  • $231k

     ...beyond traditional web indices into agent‑specific indices, model training data, live retrieval, and memory infrastructure.Create and evolve...  ...immersion in the AI ecosystem - relationships with frontier labs, agent platforms, or AI-native startups treated as a meaningful... 
    Principal
    Training
    Full time

    Expedia

    Seattle, WA
    22 hours ago
  • $151.7k - $227.6k

     ...place with Stantec.Your OpportunityAs a Principal, Healthcare Mechanical Engineering, one must...  ...with best practices, including Stantec PM Frameworks Project Involvement.Project Design...  ..., compensation, fringe benefits, job training, terminations or any other condition of employment... 
    Principal
    Training
    Full time
    Temporary work
    Part time
    Casual work
    Work at office
    Local area
    Flexible hours
    3 days per week

    Stantec

    Seattle, WA
    22 hours ago
  •  ...Amazon Web Services, Inc. seeks a Principal Solutions Architect to design cloud architectures for Foundation Model Providers, enabling training, fine‑tuning, and serving of state‑of‑the‑art models at exabyte scale. You will lead technical discussions with executives, collaborate... 
    Principal
    Training

    Amazon

    Seattle, WA
    1 day ago
  • $166k - $258k

     ...customers.Nordstrom Technology is moving to an AI Native operating model, and we're rebuilding the processes, tools, and standards our...  ...management for the operating model itself, communications, training, and adoption tracking, as its own deliverablePartner with leadership... 
    Training
    Full time

    Nordstrom

    Seattle, WA
    22 hours ago
  • $210.2k - $284.3k

     ...AWS) is seeking a world-class Principal AI Scientist to lead cutting-...  ...native approaches to network modeling, reasoning, and optimization...  ...hands‑on expertise in building, training, and evaluating AI/ML models...  ...practice, working at the frontier of AI‑Native networking with... 
    Principal
    Training
    Local area
    Flexible hours
    Day shift

    Amazon

    Bellevue, WA
    1 day ago
  • $197.9k - $267.8k

     ...software. AWS Neuron is the software stack for Trainium and Inferentia, the AWS Machine Learning...  ...strategy. Experience with ML frameworks, model architectures, and distributed computing...  ...designed to deliver the best-in-class ML training performance at the lowest training cost... 
    Principal
    Training
    Flexible hours

    Amazon

    Seattle, WA
    10 hours ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Principal PM: Frontier Model Training on Trainium. Be the first to apply!