Hybrid TPU Kernel Engineer for High-Performance ML
Neura Market
Anthropic, a public benefit corporation headquartered in San Francisco, is seeking a TPU Kernel Engineer to identify and address performance issues across ML systems, including research, training, and inference. You will design and optimize kernels for the TPU and provide feedback to researchers on how model changes affect performance, with a track record of solving large-scale systems problems and low-level optimization. #J-18808-Ljbffr Neura Market
Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the Hybrid TPU Kernel Engineer for High-Performance ML in San Francisco, CA vacancy
- Inception is seeking engineers and scientists to design, optimize, and maintain compute foundations for large‑scale language model training and inference. You will develop high‑performance ML kernels, enable efficient low‑precision arithmetic, and improve the distributed...Performance
$280k
...committed researchers, engineers, policy experts,... ...the Role As a TPU Kernel Engineer, you'll be... ...identifying and addressing performance issues across many different ML systems, including... ...experience with: High performance, large-... ...Location-based hybrid policy: Currently,...PerformanceWork at officeVisa sponsorshipFlexible hours$315k
...committed researchers, engineers, policy experts,... ...the Role As a TPU Kernel Engineer, you'll be... ...identifying and addressing performance issues across many different ML systems, including... ...experience with: High performance, large-... .... Location-based hybrid policy: Currently,...PerformanceContract workFor contractorsFor subcontractorWork at officeRelocationVisa sponsorshipWork visaFlexible hours- MakerMaker.AI in San Francisco is seeking a skilled Software Engineer to write and optimize GPU kernels. You will work on deep low-level tasks that directly impact the performance of machine learning models. The ideal candidate has over 4 years of experience with GPU kernels...Performance
- A leading AI acceleration company in San Francisco is seeking a GPU Kernel Engineer to optimize performance for machine learning models. You will be responsible for designing high-performance GPU kernels and using advanced techniques to boost computation efficiency. Ideal...Performance
$285k - $315k
SF Tensor is looking for a Founding GPU Kernel Engineer in San Francisco, specializing in GPU architecture and kernel optimization for machine... ...has deep expertise, proven capabilities in hand-optimizing performance-critical kernels, and strong programming skills in C++ and...PerformanceFull timeRelocation package- San Francisco Tensor Company is seeking a Founding GPU Kernel Engineer to enhance GPU performance for AI applications. You will optimize and write kernels while collaborating with compiler teams to improve efficiencies across architectures. The ideal candidate has deep...PerformanceWork at officeRelocation package
$75 - $85 per hour
...company to hire a Data Engineer, Consultant for its Data... ...term) Location: Hybrid — 2x/week onsite; San Diego... ...scalable, secure, and highly available data solutions... ...analysis, stabilization, performance tuning, and deployment... ...Support integration of AI/ML-ready datasets by...PerformanceContract work- ...systems. You will own the bridge between research checkpoints and production-ready inference services, designing scalable APIs and high-performance serving. You will optimize GPU workloads, manage distributed systems, and collaborate with researchers to ship demos and live...Performance
$342k
...About the RoleAs an Engineer on our hardware optimization... ...programmability and performance. You will work with our kernel, compiler and... ...needs related to ML techniques, algorithms... ...Francisco, CA. We use a hybrid work model of 3 days... ...and needs from high performance acceleratorsManage...PerformanceWork at officeLocal areaRelocation packageFlexible hours- ...generation.We're seeking a GPU Performance Engineer to squeeze every last FLOP... ...speedups. From writing custom CUDA kernels to eliminating cold start... ...custom instrumentationWrite high-performance CUDA and Triton kernels... ...utilizationCollaborate with ML engineers to optimize model...Performance
$350k
Mirendil is seeking an engineer to design and optimize custom ML kernels to enhance our model development stack. This role involves working at the intersection... ...of hardware and frontier AI research, focusing on performance optimization. The ideal candidate will have...Performance$100k - $120k
...is looking for an experienced engineer to join their founding team, focusing on low-level compute kernels to enhance robotic foundation models... ..., and familiarity with ML framework internals. Responsibilities... ...optimizations, and pioneering high-velocity development culture....Performance$132.1k - $258.4k
...As a software quality engineer at WRITER, you'll play... ...shaping the reliability, performance, and trustworthiness of... ...evolving field.This is a hybrid role based out of our... ...pipelines to ensure rapid, high-quality releases.... ...distributed systems or AI/ML applications.5+ years...PerformanceFull timeWork at officeLocal area- ...worldwide.We’re a team of engineers, clinicians, and... ...work helps care teams perform with greater precision... ...improving and integrating high performance robotic AI... ...alongside research, SW/ HW/ ML engineering, regulatory... ...: Development of Linux kernel internals, device...PerformanceLocal areaWorldwideFlexible hours
- Magic is hiring a Kernel Engineer in San Francisco to design, implement, and optimize high-performance kernels for long-context training and inference. You will tackle memory usage, data movement, and throughput challenges in real-time workloads. You’ll work across training...PerformanceVisa sponsorship
$100k - $120k
...inference workloads grow, we need kernel‑level innovations to reduce... ...team of kernel and system engineers focused on performance-critical code Design,... ...optimizations into distributed ML frameworks (e.g., PyTorch,... ...‑scale rollouts Champion a high‑velocity culture that...Performance- ...ll write and optimize the GPU kernels and supporting systems... ...This is deep, low-level work (performance counters, memory bandwidth, warp... ...actually use. We hire kernel engineers because the gap between "this... ...kernel libraries, compilers, or ML frameworks Experience with...PerformanceShift work
$285k - $315k
...re looking for a Founding GPU Kernel Engineer who lives right at the... ...hand-optimize GPU kernels for ML workloads (matmuls, attention... ...normalization, etc.) to set the performance ceilings Profile at the microarchitectural... ...) Good understanding of how high-level ML operations map to...PerformanceFull timeWork at officeRelocation package- ...and unicorn founders and senior engineers with deep expertise in 3D, generative... ...looking for a Founding Engineer, ML Inference with deep expertise in high-performance ML engineering. This is a highly... ...using torch.compile, custom CUDA kernels, and specialized inference...PerformanceRelocationVisa sponsorshipRelocation package
$285k - $315k
...future of AI and high-performance computing depends... ...We are building a Kernel Optimizer that automatically... ...with researchers, engineers, and organizations... ..., Trainium, and TPU. You'll help shape... ...options Create hybrid codegen paths for... ...Work closely with ML researchers to...PerformanceFull timeWork at officeRelocation package$180k - $280k
...investors. Since mid-2024, we've been engineering the foundation for what comes after the... ...About the role We're looking for a GPU kernel engineer with deep, low-level CUDA expertise... ...include: Write, optimize, and maintain high-performance GPU kernels (e.g., in CUDA / CuTe DSL)...PerformanceWork at officeVisa sponsorshipShift work$160k - $194k
...Francisco, CA (Remote/Hybrid Available) What is Verse... ...Role As a Software Engineer focusing on Distributed... ...transparent communication with a high caliber of mutual... ...and enhance system performance, reliability, and security... ...architectures and AI/ML landscape is a big plus...PerformanceFull timeRemote workFlexible hoursAd Verse
San Francisco, CA1 day ago$167.2k - $209k
...DigitalOcean is seeking a Senior Engineer 2 to play a key technical... ...can offer the industry-leading performance for our inference services.... ...the technical roadmap for our high-performance inference fleet.... ...the inference engine and GPU kernel layers, ensuring our infrastructure...PerformanceLocal areaRemote workWorldwideFlexible hours- ...help build the platform engineers turn to to ship AI... ...and validate networking performance on bleeding-edge clusters... ...behaviors. Optimize Kernels: You will work with communication... ...deep experience with high-performance networking... ...to a variety of ML startups, offering unparalleled...PerformanceFull timeFlexible hours
Baseten
San Francisco, CA1 day ago $130k - $200k
Join us to apply for the Founding Engineer (Systems + ML) role at Partcl . Get AI-powered advice on this job and more exclusive features... ..., synthesis. Architect and build end‑to‑end pipelines: high‑performance kernels, efficient file IO, training models, latency-sensitive...PerformanceFull time- Join a leading company in AI research as a TPU Kernel Engineer focused on optimizing performance across machine learning systems. This position is integral in designing and optimizing TPU kernels, providing feedback to improve model performance, and requires expertise in...Performance
- ...components that power OpenAI’s ML training... ...scalable, robust, and high-performance systems that maximize... ...Role As a software engineer on the Scaling team, you... ...runtimes, building custom kernels, contributing to compiler... ...Francisco, CA, with a hybrid work model (3 days/...PerformanceFull timeWork at officeLocal areaRelocation package3 days per week
OpenAI
San Francisco, CA1 day ago - Senior ML Systems Engineer, Frameworks & Tooling at Cohere Our mission is to scale intelligence... ...and hardware configurations support high-performance training. Investigate and resolve... ...distributed training libraries, or custom kernels/fused ops. Experience with multi-...PerformanceFull timeWork at officeRemote workFlexible hours
$172.5k - $260.1k
...opportunities for Lead software engineers who want their lines of... ...ourselves on writing high-quality, maintainable... ...easier. We embrace the hybrid model and celebrate the... ...with Big-Data/ML and S3 Hands-on experience... ...training, assessment of job performance, discipline,...PerformanceFull timeImmediate start
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Hybrid TPU Kernel Engineer for High-Performance ML. Be the first to apply!
Related searches
- human performance consultant San Francisco, CA
- performance test architect San Francisco, CA
- senior performance tester San Francisco, CA
- lead performance test engineer San Francisco, CA
- performance windows San Francisco, CA
- performance specialist San Francisco, CA
- performance shop San Francisco, CA
- sports performance San Francisco, CA
- performance improvement consultant San Francisco, CA
- cognitive performance San Francisco, CA