ML Research Engineer — Inference & GPU Kernels
AfterQuery
We're building a research cohort of senior domain experts in ML inference and GPU kernel engineering — spanning inference serving systems, kernel optimization, and large-model deployment infrastructure — to help evaluate how frontier AI models reason about hard, real-world expert-level problems. This is research-and-evaluation work, not production engineering: you'll be defining what "correct" and "excellent" look like on problems you already know deeply. Domain: Machine Learning / AI Experience: 6–10 years Education: PhD/Doctorate, or Master's with a strong research record Candidates who meet the requirements and provide the requested materials will be prioritized for review. As part of this process, we conduct thorough background checks. Please apply only if you meet these requirements — candidates who meet fewer than 70% of the stated qualifications may be flagged for misrepresenting their professional experience, which could affect eligibility for future project staffing.
$170.1k - $258.3k
...mobility. For the AI Kernels & Compilers team, that... ...prediction, and planning research into production‑grade... ..., and performance engineering so that every cycle on... ...builds high‑performance GPU kernels and custom libraries... ...of our on‑vehicle ML inference for ADAS and...SuggestedFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours- .... We're seeking MLOps Engineers with hands-on experience... ...any of four areas: GPU kernel programming, performance... ..., and high-throughput inference serving. This role... ...and assessing MLOps and ML systems tasks and solutions... ...to them. Guide research and engineering teams...SuggestedFull timeContract workPart timeWeekday work
- ...California. The Role: As a Research Engineer - AI Performance & Kernel Optimization , you will... ...model training and inference stacks. You will work closely... ...optimization for large-scale ML workloads, using any level... ..., HIP, Triton, or other GPU DSLs Performance tuning...SuggestedWork at officeRelocation package
$180k - $240k
...client, a venture-backed AI Startup, is hiring a talented ML/AI Research Engineer to join their team in San Francisco. The successful... ...alignment strategies across production systems. ~Optimize inference latency and GPU utilization across cloud and on-premise infrastructure....SuggestedFull time$189.6k - $237k
Scale’s ML platform (RLXF) team builds our internal distributed framework... ...language model training and inference. The platform has been powering MLEs, researchers, data scientists and operators... ...distributed ML systemsStrong software engineering skills, proficient in frameworks...SuggestedFull time$90 - $120 per hour
...technical talent with leading AI research labs. Headquartered in... .... Position: MLOps Engineer, LLM Systems (Serving, GPU Kernels, Profiling) Type:... ...profiling , debugging , and inference serving . Write accurate... ...model performance on ML systems and training...Contract workSummer workRemote work$204k - $216k
...scale. You build and run the inference infrastructure that turns... ...You sit close to applied AI, research, and platform engineering, and you make the... ...engineering to get right. The AI/ML Infrastructure and Systems... ...economics of inference, including GPU and hardware efficiency at...$141k - $249k
...training frameworks for research and production, drive... ...Waabi’s training and inference frameworks. Examples include... ...designing new CUDA kernels, quantization-aware... ...with researchers and ML engineers on best-practices for... ...in profiling CPU and GPU code using tools such...Work at officeWork from homeFlexible hours$110.7k - $379.2k
Position Summary Research Engineer — Post-Training & Small Language Models (SLMs), Healthcare... ...at scale — committed investment in GPU compute and training infrastructure, not... .... Efficient fine-tuning, training & inference infrastructure • Implement efficient fine...Local areaVisa sponsorship$200k - $225k
...AlembicAlembic is where top engineers are solving marketing's... ...systems for causal inference that maintain... ...advanced analyticsDocument research and implementation decisions... ...domainsUnderstanding of ML engineering and MLOps... ...sciencesFamiliarity with GPU computing and...$250k - $350k
...AI in production, paired with applied ML research, design, and evaluation to ensure these... ...RoleAs a Staff Machine Learning Research Engineer, you will operate across the full... ...areas as needed — training/fine-tuning, inference, memory and retrieval, evaluation and observability...Full time- ...Machine Learning Infrastructure Engineer to help architect the compute... ...integrate state-of-the-art research into production systems.... ...scale workloads across extensive GPU clusters.Build developer abstractions... ...distributed training or inference engines.Practical experience...Full timeWork at officeFlexible hours
$203.5k - $299.3k
....About the RoleWe are hiring a Causal Machine Learning Engineer to help build the causal ML foundation behind how DoorDash grows New Verticals. We... ...because you have…Deep practical experience with causal inference, econometrics, experimentation, or causal ML.Experience...Hourly payWork at officeLocal areaRemote workFlexible hours- ...Senior Machine Learning Engineer for On-Device &... ...parts of the inference stack — from a trained... ...checkpoint leaving research, through export, quantization, and kernel-level tuning, to a... ...across NPU, mobile GPU, and desktop/laptop... ...integration between the ML runtime and the...Full timeWork at officeRemote workWorldwide
$227.33k - $312.58k
We’re looking for a Staff ML Data Engineer to join Procore’s AI & Frontier... ...‑scale machine learning research and applied AI products, with... ...learning training, evaluation, or inference workflows.Solid... ...Optimizing data pipelines for GPU‑backed training and large‑scale...Full timeWork at officeLocal areaImmediate start3 days per week- ...Atlassian ecosystem.About the AI & ML Platform TeamOur team’s goal is to... ...About This RoleAs a Senior ML System Engineer on the AI & ML Platform’s Inference team, you will design and optimize... ...) to deep low-level optimizations (GPU kernels, quantization, speculative decoding...Work at officeLocal area
- ...have a legal entity. Responsibilities As a ML System Engineer on the AI & ML Platform’s Inference team, you will design and optimize large-scale model... ..., auto-scaling) to deep low-level optimizations (GPU kernels, quantization, speculative decoding).In this role,...Work at officeLocal area
$200k - $300k
...behind a useful, reliable AI agent. You will lead hands-on research and engineering that improves answer quality while managing latency, cost, and... .... What We're Looking For At least 4 years of ML or software engineering experience at an AI lab, AI product...- ...this through pioneering research in multimodal AI for... ...Research Scientist/Engineer with a focus on model... ...Strong understanding of inference performance and GPU/accelerator... ...environments Ability to read ML papers, reproduce... ...writing custom Triton/CUDA kernels or low-level...Full timeRemote workRelocation packageFlexible hours
- ...Machine Learning Engineer, Artificial Intelligence... ...the most critical ML systems in the... ...across training, inference, evaluation, and infrastructure... ...pipelines across GPU infrastructure.... ...partnership with research leadership. -... ...compilers, or GPU kernels. - Experience...Full timeRemote workWork from home
- ...Chai Discovery seeks a Research Engineer to optimize machine learning infrastructure... ...training systems, profiling GPU clusters, and ensuring... ...stack across model, layer, and kernel levels Profile end-to-end... ...industry experience with AI/ML teams on large-scale systems...
- ...are looking for an experienced ML engineer to join Flexion’s experienced... ...take ownership of Flexion’s GPU compute platforms. This is a... ...experience with the training or inference of large models (billions of... ...to never-done-before research Energetic, collaborative culture...Temporary workWork at officeRelocation
- ...to run at scale in the real world The Role You'll architect and optimize large-scale LLM workloads on GPU clusters, pushing the limits of training and inference performance for frontier models The setup Hybrid in the Bay Area, currently one day a week in person in...1 day per week
$220k - $290k
Senior / Staff Machine Learning Research Engineer Who We Are:Calico (Calico Life Sciences LLC) is... ...Engineers to join our rapidly growing ML team. You will play a critical role establishing... ...architectures to optimize training and inference performance, implementing advanced...$197.5k - $272k
...Staff Machine Learning Engineer to join our seasoned AI... ...own the end-to-end ML pipeline—from data ingestion... ...application logs, kernel traces, and multi-modalities... ...for execution on CPU/GPU-bound targets or embedded... ...C++ (C++14/17 for inference). ~ Deep proficiency...Work at officeWorldwideFlexible hoursShift work3 days per week- Title: ML Engineer Location: San Francisco, CA (Onsite) Direct HireCompany Mission Our client... ..., actuaries, and decision-makers.Research and prototype novel approaches using both... ...methods, survival analysis, or Bayesian inference is preferred.Healthcare industry experience...
$151.8k - $265.35k
...SeniorMachine Learning Engineers for our GenAI Services... ...and develop efficient inference pipelines, optimize models... ..., and mentor other ML engineers.Job ResponsibilitiesDesign... ...Collaborate with Adobe Research and other model... ...and optimize GPU-accelerated pipelines for...Full timeTemporary workLocal areaWorldwide$200k - $275k
...To learn more visit: As a Research Engineer focused on Generative AI... ...closed-loop simulation next to GPU renderers and the autonomy stack... ...from the model graph to GPU kernels and memory movement. Apply techniques... ...prototypes. You have shipped ML systems that other engineers...Full timeWork at officeWork from homeFlexible hours$193.3k - $261.5k
...member of this team, you will lead research and development efforts in... ..., and deep collaboration with engineering teams to bring research into... ...transformer architecture, training/inference lifecycles, and optimization techniques- Knowledge of ML frameworks including JAX,...InternshipLocal areaFlexible hours- ...assembling a group of senior software and ML engineers to work on one of the hardest open... ...can frontier models reason about LLM inference systems, GPU-level performance optimization, and... ...likely work on day-to-day. This is a research-and-evaluation role, not a traditional...Temporary work
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Research Engineer — Inference & GPU Kernels. Be the first to apply!
- senior ml engineer San Francisco, CA
- machine learning ai engineer San Francisco, CA
- machine learning software engineer San Francisco, CA
- ai ml engineer San Francisco, CA
- machine learning engineer San Francisco, CA
- computer vision machine learning engineer San Francisco, CA
- deep learning research engineer San Francisco, CA
- research engineer San Francisco, CA
- ai research engineer San Francisco, CA
- research programmer San Francisco, CA



