Senior Machine Learning Engineer - AI Foundation
$174.72k - $295.68kFull-time
XPENG
XPENG is a leading smart technology company at the forefront of innovation, integrating advanced AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent mobility, XPENG is dedicated to reshaping the future of transportation through cutting-edge R&D in AI, machine learning, and smart connectivity. We are looking for a full-time Machine Learning Engineer - AI Foundation, with deep knowledge and strong enthusiasm towards establishing a state-of-art ML infrastructure for training very large foundation model and accelerating model training/inference. Our mission is to solve the autonomous driving problem. You will work with a team of talented software engineers, machine learning engineers and research scientists to push the boundary of state-of-art machine learning models which will enable the next-generation E2E solution of autonomous driving. Job Responsibilities
- Optimize transformer-based LLMs for low-latency and high-throughput inference.
- Optimize kernels and model graphs using tools like CUDA, Triton, and custom fused operators.
- Implement and benchmark (Quantization, Knowledge distillation, structured and unstructured pruning, KV-cache optimization, etc.).
- Deploy optimized models across GPUs, CPUs, and edge acceleators.
- Contribute to internal tooling and documentation for model optimization flows.
- Master in CS/CE/EE, or equivalent, with 3 + years of industry experience.
- Good knowledge of PyTorch.
- Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models.
- Previous experience in the autonomous driving industry.
- Knowledge of Torchscript and Nvidia TensorRT.
- Strong programming skills in Python and C++
- Familiarity with GPU CPU, NPU, DSP architecture.
- Deep understanding of memory bandwidth, compute bottlenecks, and hardware-aware model optimization
- Being efficiently in solving complex problems collaboratively on larger teams
- A fun, supportive and engaging environment.
- Infrastructures and computational resources to support your work.
- Opportunity to work on cutting edge technologies with the top talents in the field.
- Opportunity to make significant impact on the transportation revolution by the means of advancing autonomous driving.
- Competitive compensation package.
- Snacks, lunches, dinners, and fun activities.
Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Senior Machine Learning Engineer - AI Foundation in California vacancy
$195k - $230k
...and information powered by advanced AI, recommendation systems, and adtech.... ...About the Role We are looking for a Senior Machine Learning Engineer to help evolve our large-scale... ...LLM Applications Apply LLMs and foundation models to recommendation-related problems...SeniorFoundationFull timeLocal areaWork from homeNewsbreak
Mountain View, CA21 hours ago- ...Senior Machine Learning Engineer Location: SF or Waterloo, with ability to travel Start Date: Flexible, ideally Q3 2025 About Hum.ai Hum.ai is building planetary superintelligence. Backed... ...benchmarks, and engineering foundation models that go far beyond...SeniorFoundationRemote jobFull timeWork experience placementFlexible hours
hum.ai
San Francisco, CA21 hours ago - ...care. We sit at the intersection of AI, robotics, and healthcare, operating... ...is the place. The Role As a Senior Machine Learning Engineer, you will build the intelligence layer... ...someone who knows how to turn modern foundation models into dependable production infrastructure...SeniorFoundationFull timeWork at office
Hike Medical
San Francisco, CA21 hours ago $244k - $320k
...Attentive® is the AI marketing platform for 1:... ...powered personalization engine delivers bespoke... ...Human Rights Campaign Foundation's Corporate Equality Index... ...About the Role Our Machine Learning Engineering team powers... ...thousands of brands. As a Senior Machine Learning...SeniorFoundationFull timeAttentive
San Francisco, CA21 hours ago- ...Irvine team is where embodied AI meets real robots, real... ...architectures, combining rigorous engineering with learning systems proven in globally... ...the next‑generation Field Foundation Model (FFM) , powering a... ...collaborate closely with machine learning scientists, software...SeniorFoundationLocal area
Field AI
Irvine, CA4 days ago $145k - $200k
...We are seeking a highly skilled Machine Learning Engineer with deep expertise in developing Bird’s Eye... .... Required Skills Ph.D. or Masters in AI, Computer Science, Electrical Engineering... ...in perception pipelines. Solid foundation in computer vision, deep learning, and...SeniorFoundationPlusAI, Inc.
Santa Clara, CA4 days ago- ...We are looking for a senior MLE to lead our personalization... ...implement scalable machine learning models focusing on... ...closely with other engineering teams, data scientists... ...exploring and applying AI and ML technologies. Strive... .... Solid technical foundation enabling active contribution...SeniorFoundationWork at office
Robinhood Markets, Inc.
Menlo Park, CA4 days ago $170k - $225k
...human-like skill, awareness, and learning capabilities. Our systems are... ...reality! About the Role As a Senior/Staff Machine Learning Engineer, you will be working on a myriad... ...develop Dexterity’s technology foundations in Physical AI. Additionally, you will be...SeniorFoundationFull timeWorldwideDexterity
Redwood City, CA21 hours ago$180k - $300k
...Senior Software Engineer, Machine Learning We’re looking for a product-minded machine learning engineer who’s excited... ...boundaries of what’s possible with AI. At Arcade, you’ll experiment with... ...brand personality — crafting the foundation for content that builds and...SeniorFoundationRemote workRelocationArcade
San Francisco, CA2 days ago$201.68k - $322.69k
...allow you to hone existing skills and learn new ones Machine Learning Engineer at Capital Group We are seeking a... ..., build, and operate Generative AI systems that improve Capital Group'... ...production systems with solid architectural foundations. Strong understanding of system...SeniorFoundationTemporary workLocal areaFlexible hoursCapital Group
Irvine, CA4 days ago- ...the intersection of AI, biology, chemistry, and large-scale engineering. Our goal is to... ...scientific data into machine intelligence capable... ...not only strong data foundations but also carefully designed learning systems that can... ...The Role As a Senior AI/ML Engineer, you...SeniorFoundationFull timeRemote workFlexible hours
Absentia Labs
San Francisco, CA21 hours ago $214k - $289.5k
Overview Come join Intuit as a Senior Staff Machine Learning Engineer (MLE). Senior Staff MLEs deliver end-to-end AI solutions that span multiple domains and products, influencing... ...transformative technologies such as foundation models, retrieval-augmented generation (RAG...SeniorFoundation$345.04k - $399.42k
...Senior Machine Learning Engineering Manager Every day, tens of millions of people come to Roblox to explore... ...experiences for everyone. Why Safety AI Systems? As Senior Engineering Manager... ...with large language models or other foundation model technologies. Expertise in...SeniorFoundationFull timeWork experience placementWork at officeLocal areaMonday to FridayRoblox
San Mateo, CA3 days ago$204k - $259k
...team builds the system which learns the spatial-temporal representation... ...set of sensors, enabling engineers like you to (1) develop... ...pipelines for cutting-edge VLM foundation models. Conduct research and... ...~5+ years of experience in Machine Learning, with a focus on large...SeniorFoundationFull timeRemote workWaymo
San Francisco, CA21 hours ago- ...Possible. At Pinterest, AI isn't just a feature,... ..., we’ll explore your foundational skills and how you... ...multimodal representation learning, heterogeneous graph... ...is a small group (~6 engineers along with a product prototyping... .... ~ M.S. or PhD in Machine Learning, Computer...SeniorFoundationFull timeCurrently hiringWork at officeRemote workRelocationRelocation package
Pinterest
San Francisco, CA21 hours ago $281k - $356k
...future innovations. You'll build active learning and ML-aided labeling workflows to... ...You'll partner closely with Product and Engineering teams, especially those focused on data... ...large-scale vision, video, or multi-modal foundation models. Familiarity with end-to-end...SeniorFoundationFull timeTemporary workRemote workWaymo
Mountain View, CA21 hours ago$298k - $368k
...team builds the system which learns the spatial-temporal representation... ...set of sensors, enabling engineers like you to (1) develop... ...~7+ years of experience in Machine Learning, with a focus on large... ...development (LLM, VLM, or similar foundation models). ~ Proven expertise...SeniorFoundationFull timeRemote workWaymo
San Francisco, CA21 hours ago$70k - $200k
...Who are We? Field AI is transforming how robots interact with the... ...through real-field applications. Learn more at . About the Job Our Field Foundation Model (FFM) powers a global fleet... ...insights that empower our customers and engineers to deliver value. Field-insight...SeniorFoundationFull timeRemote workField AI
Irvine, CA21 hours ago- ...it to train multimodal foundation models that learn the context-dependent nature... ...combining cutting-edge machine learning with the most... ...what’s possible. As a Senior Machine Learning Engineer , you will play a... ...art techniques to train AI virtual cell models on our...SeniorFoundationWork at officeVisa sponsorshipFree visa3 days per week
$160k - $220k
...leading provider of AI-powered cybersecurity... ...approach that automatically learns and adapts to dynamic... ...is looking for a Sr. Machine Learning Engineer to aid in building... ...cybersecurity. As a Senior Machine Learning... ...Strong mathematical foundation (e.g., degree in Computer...SeniorFoundationHourly payWork at officeLocal areaRemote workFlexible hours$165k - $225k
...recruiting for one of its clients a Senior Machine Learning Engineer - this is a fully remote role for US/... ...adoption worldwide through cutting-edge AI-powered technology. Our solutions... ...datasets to train, validate, and fine-tune foundation models and custom architectures...SeniorFoundationRemote workWorldwide- ...high-performance deep learning models that generate dense... ...sensor modalities . Engineer temporal processing... ...Computer Science, Robotics, Machine Learning, or related... ...Model, or multi-modal 3D foundation model, or World Model,... ...intelligence (AI) tools to support parts...SeniorFoundationTemporary workRelocation package
$178.4k - $267.6k
...Technologies, Inc. Job Area: Engineering Group, Engineering Group Machine Learning Engineering General Summary:... ...edge. We are part of Qualcomm AI Research, and we focus on advancing... ...skills combined with a solid foundation in AI and general ML techniques....SeniorFoundationWork experience placementWork from homeQualcomm
San Diego, CA21 hours ago- ...Description Job Description As a machine learning engineer within the Attributes team in the Perception... ...models you develop will serve as the foundation for how our vehicles understand and... ...We may use artificial intelligence (AI) tools to support parts of the hiring...SeniorFoundationTemporary workRelocation package
$200k - $280k
...work done. Our Physical AI and fleet services... ...group of bold thinkers—engineers, innovators, and industry... ...give our autonomous machines human-like awareness in... ...vision and machine learning systems that turns noisy... ...scene representations, foundation-model-driven data...SeniorFoundationFull timeTemporary workFlexible hoursShift work- ...Intelligence team is responsible for developing Foundation Models for ML Agents and planning,... .... As an ML Agents and Planning Machine Learning Engineer you will work on the bleeding edge of... ...We may use artificial intelligence (AI) tools to support parts of the hiring...SeniorFoundationTemporary workRelocation package
- ...trajectory. As a Principal ML Engineer, you will lead the technical development of machine learning algorithms for our next-generation... ...with teams specializing in Foundation Models, Perception, Simulation,... ...technical leadership in large-scale AI organizations Significant...SeniorFoundationTemporary workImmediate startRelocation package
$180k - $220k
...Ouster, we build sensors and tools for engineers, roboticists, and researchers, so they... ...We are looking for a highly technical Machine Learning Engineer to lead our efforts in... ...or Few-Shot Learning. Mathematical Foundation: Strong grasp of linear algebra and probability...SeniorFoundationFull timeWork experience placementLocal areaOuster
San Francisco, CA21 hours ago$225k - $300k
...Machine Learning Engineer About Latent Health Healthcare today is only... ...training and post-training of foundation models Novel objectives... ...We are primarily hiring for senior and staff-level engineers... ...Build systems that define how AI is trusted in clinical...FoundationFull timeWork at officeImmediate startLatent
San Francisco, CA21 hours ago- ...development, and deployment of advanced AI agents and agentic systems. Architect and... ...managers, UX designers, and other engineers to define requirements and deliver impactful... ...command execution. Knowledge and passion in machine learning algorithms, GenAI, LLMs, and Agentic AI...SeniorFull timeWork experience placement
Eightfold
Santa Clara, CA21 hours ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior Machine Learning Engineer - AI Foundation. Be the first to apply!
Related searches
- machine learning engineer California
- computer vision machine learning engineer California
- senior manager quality engineering California
- senior device engineer California
- senior platform engineer California
- senior application developer California
- senior implementation project manager California
- senior director epidemiology California
- sr finance manager California
- senior commercial counsel California