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
$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...TrainingTemporary workWork experience placement- 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
- 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
- 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
- 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
- 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
- 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
- ...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
- ...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
$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...TrainingFull time- ...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
- ...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...TrainingShift work
- ...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
- ...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....
- ...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
- ...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
- 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
$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)...TrainingWork at office3 days per week$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...TrainingTemporary workWork at officeFlexible hours3 days per week$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...TrainingFull timeContract workLocal area$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...TrainingVisa sponsorship- 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
$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$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...TrainingWorldwideShift work$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- 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...
$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...TrainingWorldwide- ...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
$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$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...TrainingFull timeRemote work
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!
- lead infrastructure engineer Palo Alto, CA
- lead operating engineer Palo Alto, CA
- lead network engineer Palo Alto, CA
- lead engineer Palo Alto, CA
- senior ml engineer Palo Alto, CA
- machine learning ai engineer Palo Alto, CA
- machine learning software engineer Palo Alto, CA
- ai ml engineer Palo Alto, CA
- machine learning engineer Palo Alto, CA
- computer vision machine learning engineer Palo Alto, CA



