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

Lead ML Platform Engineer for Scalable GPU Training

Aumni

JPMorganChase in Palo Alto seeks a Lead Software Engineer to architect and operate scalable ML training platforms on AWS and other clouds. You will optimize GPU workloads, enable Gen AI workflows, and enforce governance with observability and automation across environments. The role emphasizes enterprise-grade security, reproducibility, and efficient collaboration with data engineering and platform teams. A strong Linux+Python background and experience in cloud ML are essential. #J-18808-Ljbffr Aumni

Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Lead ML Platform Engineer for Scalable GPU Training in Palo Alto, CA vacancy
  • $180k

     ...motivated, and focused on engineering excellence. This...  ...THE ROLE: As an ML Infrastructure...  ...high-performance ML platform that powers recommendations...  ..., and scaling GPU compute infrastructure, training frameworks, and...  ...stack Ensuring scalability, reliability, and efficiency... 
    Training
    Temporary work
    Work experience placement

    Maven Ventures

    Palo Alto, CA
    1 day ago
  • JPMorganChase is seeking a Lead Software Engineer in the AI/ML Data Platforms group to design, build, and operate scalable ML training systems on AWS and Kubernetes. You will drive GPU-based training pipelines, performance optimizations, and governance across environments... 
    Training

    JPMorgan Chase & Co.

    Palo Alto, CA
    1 day ago
  • JPMorganChase in Palo Alto is seeking a Lead Software Engineer to drive the design, build, and operation of scalable ML training systems on AWS and other clouds. You will optimize GPU workloads, enable Gen AI/LLM...  ...with data engineering and platform teams to define interfaces,... 
    Training

    JPMorgan Chase & Co.

    Palo Alto, CA
    1 day ago
  • JPMorganChase in Palo Alto seeks a Lead Software Engineer to design, build, and operate scalable ML training systems on AWS and cloud platforms. You will drive end-to-end training pipelines, run GPU workloads, and ensure governance and observability across environments.... 
    Training

    JPMorganChase

    Palo Alto, CA
    1 day ago
  • Mistral AI is seeking an experienced engineer to build and operate the ML platform powering large-scale training, evaluation, and batch inference. You will develop infrastructure...  ...researchers and engineers to run distributed GPU workloads across clusters, hardware types, and... 
    Training

    Precision Labs

    Palo Alto, CA
    5 days ago
  • JPMorganChase in Palo Alto is seeking a Software Engineer III to join the AI/ML data platform team. You will design and operate scalable ML training systems on AWS and other clouds, productionizing training workloads, and improving performance and cost efficiency. You will... 
    Training

    JPMorgan Chase

    Palo Alto, CA
    1 day ago
  • Apple is looking for a Senior Engineer to join the ML Compute Team in Santa Clara...  ...workloads including training and inference, collaborating across Apple to deliver scalable backend infrastructure. You...  ...tackles large, complex ML platforms. This role emphasizes reliability... 
    Training

    Apple Inc.

    Santa Clara, CA
    1 day ago
  •  ...take your software engineering career to the next...  ..., stable, and scalable technology products...  ...and skills Formal training or certification on...  ...with emphasis on ML systems. Hands-on...  ...more public cloud platforms (AWS, GCP) Experience...  ...deploying or operating GPU workloads in... 
    Training

    Aumni

    Palo Alto, CA
    1 day ago
  •  ...THE ROLE:We are hiring AI / ML Platform Engineers to build the platform layer...  ...AI-for-engineering workflows scalable, reliable, and reproducible....  ...agent execution, distributed training and inference, experiment...  ...automation, artifact management, and GPU cluster utilization.You will... 
    Training

    AMD

    Santa Clara, CA
    1 day ago
  • $272k - $431.25k

     ...a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to...  ...potent, effective, and scalable solutions as we mold...  ...bottlenecks and lead initiatives to systematically...  ...distributed training operations using PyTorch...  ...cloud computing platforms (e.g., AWS, GCP, Azure... 
    Training
    Full time

    Nvidia

    Santa Clara, CA
    2 days ago
  •  ...an early career machine learning engineer to help build and deploy state-of-the-art ML models for the electric grid. You...  ...AI. You will design and implement scalable ML pipelines, deploy models to production, and containerize training code for cloud execution, partnering... 
    Training

    X Development, LLC

    Mountain View, CA
    1 day ago
  •  ...seeking a Machine Learning Engineer to help build the platform behind our product intelligence, production ML models, and the teams that...  ...feature pipelines, distributed training, and model serving at scale...  ...to impact reliability and scalability across the #J-18808-Ljbffr... 
    Training
    Shift work

    Unity

    Mountain View, CA
    21 hours ago
  •  ...criticalproblemsfor an industry leading financial institution....  ..., while raising engineering and scientific...  ...success metrics, and scalable solution designs Build...  ...and Skills Formal training or certification on applied...  ...machine learning platforms, such as Amazon Web Services... 
    Training

    JPMorgan Chase & Co.

    Palo Alto, CA
    1 day ago
  •  ...Mountain View, California, is seeking a Machine Learning Engineer to design and optimize machine learning models and...  ...equivalent. Key responsibilities include developing scalable ML pipelines, optimizing existing platforms, and researching new data management techniques.... 

    Samsung

    Mountain View, CA
    1 day ago
  •  ...building the future of Ford’s autonomous platform with scalable ML and cloud infrastructure in...  ...data pipelines, and run distributed training across teams to advance autonomous driving...  ...The role requires a degree in Computer Engineering/CS/EE/Robotics with 4+ years of relevant... 
    Training

    Latitude

    Palo Alto, CA
    4 days ago
  •  ...a Senior Machine Learning Engineer, Data Infrastructure, to design...  ...evolve the offline data platform that enables model training and experimentation at...  ...Airflow. You will partner with ML engineers to improve...  ...decisions to keep pipelines scalable and robust. #J-18808-Ljbffr... 
    Training

    LE130 Unity Technologies SF

    Mountain View, CA
    2 days ago
  • Rivian is seeking a Staff Software Engineer, ML Data Infrastructure to own how autonomy...  .... You’ll collaborate with AI Platform, Perception, Planning, Simulation teams...  ...engineering groups to enable scalable analytics and model training on multimodal drive data from #J-1... 
    Training

    Rivian

    Palo Alto, CA
    1 day ago
  • $250k - $350k

     ...Staff level Inference Engineers to accelerate the performance...  ...acceleration, GPU parallelism, advanced...  ...systems deliver industry-leading user experiences at...  ...maximal efficiency and scalability.Programming for Performance...  ...production.Improve Training Efficiency: (Bonus)... 
    Training
    Work at office
    3 days per week

    Pika

    Palo Alto, CA
    1 day ago
  • $140k - $230k

     ...our software development platform for software-defined...  ...accelerators, deploying scalable and efficient machine learning (ML) training and evaluation pipelines...  ...for a skilled Software Engineer or Machine Learning Engineer...  ...SIMD/SIMT parallelism, GPU programming, multithreading... 
    Training
    Temporary work
    Work at office
    Flexible hours
    3 days per week

    Woven by Toyota

    Palo Alto, CA
    28 days ago
  • $228k - $285k

     ...a Staff Software Engineer for ML Optimization and...  ...Acceleration, you will be a lead member of the...  ..., including Post-Training Quantization (PTQ...  ...on embedded platforms. Performance...  ...understanding of GPU architectures to...  ...targets, ensuring scalability across the Rivian... 
    Training
    Full time
    Contract work
    Local area

    Rivian

    Palo Alto, CA
    1 day ago
  • $250k - $350k

     ...alongside some of the world's leading ML systems engineers, including leaders behind...  ...powering our large-scale training, inference, and reinforcement...  ...to maximize performance, scalability, reliability, and productivity...  ...Optimize memory, GPU kernels and communication for... 
    Training
    Visa sponsorship

    Periodic Labs

    Menlo Park, CA
    21 hours ago
  • LinkedIn is seeking a Staff Software Engineer to join the AI Training Infra team in a hybrid role, working...  ...teammates while advancing large-scale AI platforms. The role emphasizes distributed systems, GPU‑based inference, and scalable data pipelines across countless... 
    Training

    LinkedIn

    Mountain View, CA
    21 hours ago
  • $138k - $200k

     ...expanded, or accelerated TPU/GPU partnerships and...  ...strategy.Experience leading agreements, cross-functional...  ...industry, promoting Platform-as-a-Service (PaaS) or...  ..., capacity, product, engineering, and finance teams to...  ...relevant education or training. US: $138000 - $200000... 
    Training

    Google

    Sunnyvale, CA
    4 days ago
  • $171k - $231.5k

     ...financial technology platform that powers...  ...the Fintech Risk AI/ML credit risk team(Team...  ...Senior Machine Learning Engineer. Team Apollo is...  ...build pipelines to train and deploy modelsPartner...  ...deploying highly scalable software supporting...  ...with GPU acceleration (i.e CUDA... 
    Training
    Worldwide
    Shift work

    Intuit

    Mountain View, CA
    4 days ago
  • $150k

     ...economy. The Role The GPU Kernel Engineer will play a role at the forefront...  ...stacks, especially at training and inference, and support the...  ...-art hardware and software platforms to improve their efficiency...  ...algorithms Participate in, or lead design reviews with peers... 
    Training

    Institute of Foundation Models

    Sunnyvale, CA
    4 days ago
  • JPMorgan Chase’s Software Engineer III role sits within the Commercial and Investment Banking...  ...design, develop, and deploy secure, scalable ML-powered software solutions, with emphasis...  ...cloud/Kubernetes expertise, including GPU workloads and modern microservices. Strong... 

    Testing Site - JPMC

    Palo Alto, CA
    21 hours ago
  • $207k - $300k

     ...technical proficiency ML workload...  ...-level ability to lead complex, multi-quarter...  ...Google's software engineers develop the next-generation...  ...a principled, scalable methodology to...  ...providing the essential platforms that enable...  ...relevant education or training. US: $207000 - $30... 
    Training
    Worldwide

    Google

    Mountain View, CA
    1 day ago
  •  ...framework, our flagship platforms - Nace Data...  ...Machine Learning Engineer, you will play a central...  ...research into scalable, production-ready...  ...opportunities where ML can drive product...  ...pipelines, model training, debugging, and performance...  ...with multi-node GPU training. ~... 
    Training

    Nace AI

    Palo Alto, CA
    21 hours ago
  • $124k - $250k

     ...LifeAs a member of our software engineering infra team, you'll solve...  ...globally distributed ecosystem platform of services that in turn...  ...for providing and maintaining scalable infrastructure with high...  ...of model delivery, including training, serving, and optimizations,... 
    Training

    AppLovin

    Palo Alto, CA
    5 days ago
  • $213k - $263k

     ...range of vehicle platforms and product use...  ...for testing and training the Waymo Driver...  ...group of software engineers, machine learning (ML) engineers, and...  ...industry-leading simulation solutions...  ...transfer research into scalable and production-...  ...scale models on GPU/TPU clusters are... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Lead ML Platform Engineer for Scalable GPU Training. Be the first to apply!