Staff ML Frameworks Engineer - ML Infra & Runtimes
Google is seeking a Staff Software Engineer for ML Frameworks in Mountain View, CA. You will own outcomes for ML infrastructure, shape technical direction, and drive end‑to‑end deployment of experiments and systems across TensorFlow, JAX, and PyTorch ecosystems. You will influence stakeholders, navigate ambiguity, and lead from design through deployment, ensuring scalable, efficient ML platforms for Google and Cloud developers alike. #J-18808-Ljbffr Google
Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Staff ML Frameworks Engineer - ML Infra & Runtimes in Mountain View, CA vacancy
$159.05k - $199.3k
About the role We are looking for a software engineer with deep experience in optimizing ML models and deploying them on production‑grade embedded runtime environments. You’ll work across the entire ML framework stack (e.g. PyTorch, JAX, ONNX, TensorRT, CUDA, XLA, Triton...SuggestedFull timeFor contractorsFor subcontractor- Apple’s Server ML Frameworks team in Cupertino seeks engineers to optimize high-performance GenAI applications. This role focuses on ML inference framework projects and GPU programming while collaborating with cross-functional teams to enhance performance. The ideal candidates...Suggested
$153.2k - $234.1k
...seeking a talented professional to lead the design and implementation of scalable machine learning platforms as part of their Embodied AI Infra Foundation team. The ideal candidate will have over 3 years of experience in building distributed systems, strong coding skills in...SuggestedRemote jobWork at office3 days per week- ...Mountain View is seeking an experienced professional to enhance ML infrastructure for training workloads. Responsibilities include... ...strong background in Python and C++, as well as experience with ML frameworks like TensorFlow. This role offers competitive salary and...Suggested
$213k - $263k
...across 15+ U.S. states. The ML Optimization team at Waymo provides... .... We are looking for engineers with ML software & systems expertise... ...to the Senior Manager of Runtime and Optimization. You will:... ...and major deep learning frameworks (e.g., PyTorch, JAX). Experience...SuggestedFull timeRemote work- Orbifold AI is seeking a Machine Learning Engineer in Palo Alto, California, to build and optimize the infrastructure for multimodal... ...proficiency in Python, and proven knowledge of machine learning frameworks, particularly PyTorch. You will directly influence the...
- ...Generative AI/LLM's + knowledge graph + metadata lake Experience: 6+ years industry overall experience with 3+ in ML Infra or MLE Expertise: back end software engineering strength with recent industry exp making ML systems more reliable/scalable (with opportunities to help...Full timeImmediate start
- ...Machine Learning/Reinforcement Learning Infrastructure Engineer to design scalable, reliable infrastructure for AI... ...ideal candidate will have substantial experience in ML infrastructure, expertise in training frameworks like PyTorch and JAX, along with proficiency in...
- ..., and hyper-connected world. Role Overview As our Staff Software Engineer, ML infra Engineer for Search & Discovery organization, you will be... ...memory tuning) Experience with Big Data tools and ETL frameworks (Hadoop, Hive, Presto, Spark, Scala, Apache Airflow, etc...Temporary workFlexible hours
- ...company is seeking a Senior Machine Learning Engineer in Mountain View, California. This role... ...teams and ensure the scalability of ML systems. The ideal candidate has over 5 years... ...with expertise in Python, deep learning frameworks like PyTorch, and experience in building...
- ...unique intersection of research, software engineering, hardware engineering, and product... ...and acceleration, building ML compilers and runtimes for efficient execution, and creating... ...). Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.). Experience...
$153.2k - $234.1k
General Motors is seeking a Senior ML Infra Engineer in Sunnyvale, California, to develop and implement machine learning solutions for autonomous vehicles. You will design scalable platforms, contribute to technical discussions, and mentor junior engineers. Candidates should...Remote job$218.8k - $335.3k
Job Description Staff AI/ML Engineer, AV ML Infra We're General Motors (GM), a company driving the future of mobility with advanced self-driving and electric vehicle technologies. We're building the world's most innovative autonomous vehicles to safely connect people to...Local areaWork from homeFlexible hours$153k - $222k
Applied Intuition in Sunnyvale, California is looking for ML Engineers to design and implement training approaches for deep learning. You'll help build scalable machine learning pipelines and work collaboratively across teams. This role demands a Bachelor's degree in Computer...- Google is seeking a Software Engineering leader in Sunnyvale to develop and drive ML infrastructure and tools supporting Google's products like Search and YouTube. The ideal candidate will have extensive experience and technical expertise in creating scalable ML environments...
- Nuro is seeking a Software Engineer in Mountain View, CA, focused on building ML Infrastructure. You will design systems for automated resource provisioning, workload scheduling, and data processing to enhance the Nuro Driver™ development lifecycle. The ideal candidate...
- Apple is looking for an ML Infrastructure Engineer in Cupertino, CA, to develop APIs that enhance the experience of machine learning on its platforms... ...demands proficiency in Python, solid experience with ML frameworks like PyTorch, and a good understanding of ML fundamentals....
$238k - $302k
Waymo is seeking engineers with expertise in ML systems to join their team in Mountain View, California. In this role, you will design and build distributed evaluation platforms for large-scale ML evaluation workloads, collaborate with researchers and engineers, and optimize...Full time$128.7k - $261.3k
...development, and performance engineering so that every cycle on our accelerators... ...reliable, and effortless for ML engineers across the AV... ...their integration with our runtime—so that we can simultaneously... .... Experience with ML frameworks (e.g., PyTorch, TensorFlow, JAX...Local areaFlexible hours- AppLovin in Palo Alto is searching for a ML Infrastructure Engineer to design and optimize large-scale distributed systems. You will collaborate with engineering and research teams to tackle various technical challenges, directly impacting product technology and performance...
- Palantir is seeking a software engineer to join its team focused on enabling machine learning models in production. You will work on deploying AI models across various environments, including government networks and enterprises with strict data sovereignty. Your work will...
- An innovative AI startup is seeking a Founding ML Infrastructure Engineer to take charge of deploying and optimizing production-grade LLM systems. In this core role, you will be responsible for building and managing a full ML serving stack, working closely with product...
$173k - $307k
...Applied Science & Machine Learning ( ASML ) is seeking a Staff Machine Learning Training Framework Engineer to play a critical role in building and scaling the... ...that support multiple users or teams. Proven ML Training Experience: Hands‑on experience training models...Temporary workLocal areaWorldwide- Waymo is seeking an ML Infrastructure Engineer to design and optimize distributed input data pipelines... ...engineers to remove bottlenecks, boost runtime goodput, and ensure scalability across... ...profiling performance and contributing to ML infra improvements. #J-18808-Ljbffr Waymo
- ...position has a strong technical background, hands-on software engineering experience, and a knack for solving hard problems. This role is... ...our Mountain View, CA office! What you'll do: Own development of ML models end-to-end from data strategy, initial development,...Work experience placementWork at office
- A major automotive company in Sunnyvale, California is seeking a Staff ML Infrastructure engineer to enhance AI validation efforts. This role focuses on building and scaling compute platforms for simulation and data workflows. Ideal candidates should have extensive experience...
$140k - $252k
Tesla Motors, Inc. is seeking an ML/RL Infra Engineer in Palo Alto, California, to build scalable infrastructure for agentic models. The role... ...have expertise in distributed systems and machine learning frameworks like PyTorch. Along with a competitive salary range of $140...$213k - $263k
...are looking for researchers and software engineers who are passionate about developing machine... ...large distributed systems covering the ML lifecycle, supporting planet-scale dataset... ...to machine learning tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar...Full time$272k - $431.25k
...be doing: Engage closely with our AI and ML research teams to discern their infrastructure... ...of teams, such as researchers, data engineers, and DevOps professionals, to develop a cohesive... ...developments in AI/ML technologies, frameworks, and successful strategies, and advocate...NVIDIA
Santa Clara, CA3 days ago$213k - $263k
Waymo is seeking experienced engineers with ML software and systems expertise to develop the next generation of its onboard ML inference engine. The role involves architecting high-performance ML systems for autonomous vehicles, requiring over 5 years of software engineering...Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Frameworks Engineer - ML Infra & Runtimes. Be the first to apply!
Related searches
- technology administrator Mountain View, CA
- engineering aide Mountain View, CA
- senior staff systems engineer Mountain View, CA
- staff data engineer Mountain View, CA
- software engineer staff Mountain View, CA
- senior staff engineer Mountain View, CA
- assistant engineer Mountain View, CA
- staff engineer Mountain View, CA
- data scientist machine learning engineer Mountain View, CA
- machine learning engineer Mountain View, CA