Member of Technical Staff - Kernels & GPU Performance
Gimlet Labs
About Us Gimlet is building the next generation of AI infrastructure: large-scale AI datacenters and the orchestration platform that coordinates them. The future of AI will require vastly more compute than exists today. But as AI workloads become more complex and new hardware architectures emerge, simply deploying more GPUs isn't enough. The challenge is making increasingly diverse compute work together. Gimlet's platform intelligently partitions and routes workloads across heterogeneous hardware, enabling step-function improvements in performance and efficiency. Customers deploy through production-grade APIs without needing to think about hardware selection, placement, or optimization. We work with foundation labs, hyperscalers, and AI-native companies to power production workloads at massive scale and help define the infrastructure layer for the future of AI.
About the role Gimlet Labs is seeking a Member of Technical Staff focused on kernels and GPU performance. In this role, you will work close to accelerators and execution hardware to extract maximum performance from AI workloads across diverse and rapidly evolving platforms. You will analyze low-level execution behavior, design and optimize kernels, and ensure performance is reliable across both established and emerging hardware. This role is ideal for engineers who enjoy deep performance work, reasoning about hardware tradeoffs, and turning theoretical peak performance into real-world results.
What you will work on
About the role Gimlet Labs is seeking a Member of Technical Staff focused on kernels and GPU performance. In this role, you will work close to accelerators and execution hardware to extract maximum performance from AI workloads across diverse and rapidly evolving platforms. You will analyze low-level execution behavior, design and optimize kernels, and ensure performance is reliable across both established and emerging hardware. This role is ideal for engineers who enjoy deep performance work, reasoning about hardware tradeoffs, and turning theoretical peak performance into real-world results.
What you will work on
- Design, implement, and optimize GPU and accelerator kernels for AI workloads
- Analyze and tune performance across the GPU execution stack, including memory access patterns, synchronization, and instruction scheduling
- Work with compilers and runtimes to ensure kernels integrate cleanly and perform well in end-to-end systems
- Bring up and optimize execution on new or emerging accelerators
- Profile, benchmark, and debug performance issues across kernels, runtimes, and hardware
- Ensure performance optimizations are robust, correct, and production-ready at scale
- Strong software engineering fundamentals
- Experience working on performance-critical systems close to hardware
- Comfort reasoning about low-level execution behavior, memory hierarchies, and performance tradeoffs
- Experience with CUDA, Triton, CUTLASS, or other accelerator programming models
- Deep understanding of GPU execution models (warps/wavefronts, blocks, grids)
- Experience optimizing memory access patterns (coalescing, shared memory, cache behavior)
- Familiarity with occupancy, latency hiding, and instruction-level parallelism
- Experience using profiling and performance analysis tools
- Familiarity with multi-GPU or distributed execution is a plus
Vacancy posted 23 hours ago
Similar jobs that could be interesting for youBased on the Member of Technical Staff - Kernels & GPU Performance in San Francisco, CA vacancy
$225k
...compute to achieve this goal. About The Role As a Kernel Engineer, you will design, implement, and maintain high-performance kernels to optimize throughput and latency... ...Blackwell or Google TPUs Develop and optimize GPU kernels in frameworks such as NCCL, MSCCLPP, CUTLASS...PerformanceRelocationVisa sponsorship- ...reliably than humans can alone. Our technical approach combines frontier-... .... About the Role As a Kernel Engineer, you will design, implement, and maintain high-performance kernels that optimize throughput... ...Experience developing and optimizing GPU or accelerator kernels using...PerformanceWork at officeVisa sponsorshipRelocation packageFlexible hours
- ...challenges and the wins. What You'll Do Bring deep kernel expertise to our AI agents that optimize high-performance, mission-critical computing systems. You'll shape... ...or optimizing kernels for ML or other GPU-heavy workloads Fluency in Python and C/C++, and...PerformanceWork at officeFlexible hours
- ...Member Of Technical Staff - Image / Video Generation Freiburg (Germany) About... ...models don't fit on one GPU and training decisions impact... ...and backward Triton kernels and ensuring their correctness... ...trace viewers Know the performance characteristics of different...PerformanceRemote workWorldwide2 days per week
- Member of Technical Staff, ML Infrastructure & Inference Overview We are a cutting... ...utilization and better performance across multi-vendor systems... ...systems Partner with compiler, kernel, networking, and... ...Serving, Distributed Systems, GPU Infrastructure, AI Infrastructure...Performance
- ...Member of Technical Staff, Model EfficiencyWho are we?Our mission is to scale intelligence... ...inference stack to improve core performance metrics by diving deep into model... ...performance techniques, including GPU/CUDA optimizations, kernel-level improvements, and model execution...PerformanceFull timeWork at officeRemote workFlexible hours
- ...With a focused team, breakthrough performance doesn't require breakthrough compute... ...that matter, and join the team. As a Member of Technical Staff with a focus on Multimodal AI, you will... ...: Experience in writing efficient GPU kernels using CUDA, optimising performance...PerformanceFull timeWork at officeRemote workFlexible hours
$150k - $300k
...training stack. Core Technical Responsibilities LLM Serving... ...across our cloud GPU fleets. GPU‑Aware... ...Inference Optimization & Performance Framework Development... ...Performance: Profile kernels, memory bandwidth and... ...development and encourage team members to contribute to the...PerformanceWork at officeRemote workVisa sponsorshipRelocation packageFlexible hoursShift work$150k - $300k
...distributed system with performance engineering at its... ...skills, from deep Linux kernel topics to high-level distributed... ...at scale. Core Technical Responsibilities... ...heterogeneous hardware (CPU, GPU, TPU) Platform... ...development and encourage team members to contribute to the...PerformanceWork at officeRemote workVisa sponsorshipRelocation packageFlexible hours- ...enabling step-function improvements in performance and efficiency. Customers deploy... ...the role Gimlet Labs is seeking a Member of Technical Staff focused on ML systems and inference.... ...boundaries Work closely with compilers, kernels, networking, and distributed systems...Performance
$225k
...large-scale model training across massive GPU clusters. You will work at the boundary... ...systems, ensuring that training runs are performant, reliable, and reproducible under extreme... ...training throughput Collaborate with Kernels and Research to align model architecture...PerformanceRelocationVisa sponsorship$256k - $276k
...Postman. The Opportunity As a Member of Technical Staff on AI Infrastructure, you will build and... ...and research teams to ensure performance, scalability, and reliability of critical... ...services Optimize performance for GPU/xPU accelerators and cloud environments...PerformanceWork at officeFlexible hours3 days per week$150k - $300k
...fine-tuning runs on managed GPU clusters with a single API call... ...runs the jobs. Core Technical Responsibilities Hosted... ...fundamentals: networking, namespaces, performance tuning Programming &... ...and encourage team members to contribute to the broader...PerformanceWork at officeLocal areaRemote workVisa sponsorshipRelocation packageFlexible hours- ...efficiency Dataloaders, fusion, activation remat, gradient checkpointing. FSDP/ZeRO/tensor+pipeline parallel; NCCL tuning. GPU + kernel performance Nsight profiling, Triton/CUDA kernels, fused ops. Flash-attention-style speedups, sequence packing, KV-cache tricks....Performance
- ...optimize distributed inference systems on GPU clusters, pushing throughput with large-... ...efficient low-level code (CUDA, Triton, custom kernels) and integrate it seamlessly into high-... ...systems, ML infrastructure, or high-performance serving (8+ years) Production-grade expertise...PerformanceRemote job
- Member of Technical Staff, ML Systems Mirendil Mirendil is a tech-first company focused on solving core bottlenecks that unlock step... ...models Designing and implementing custom ML kernels Optimizing performance (latency, throughput, cost) Developing data pipelines...Performance
- ...the foundation model training stack, from data pipelines to GPU kernels Design, build, and optimize distributed training systems (PyTorch... ...tools for large-scale runs, enabling rapid diagnosis of performance regressions and failures What You’ll Bring Deep experience...PerformanceRemote job
- ...optimizes AI itself. Our journey starts with GPU kernels, but will expand into every corner of... ...systems that help the agent diagnose performance bottlenecks Ship features that... ...You're a strong fit if you: Have deep technical intuition and can learn new domains quickly...PerformanceRemote work
- ...component to hardware that best fits its performance and efficiency needs. This approach... .... Gimlet Labs is seeking a Member of Technical Staff focused on compilers. In this role, you... ...spanning graph-level, tensor-level, and kernel-level representations Implement partitioning...Performance
$225k
...reliable. What you’ll work on Design and scale high‑performance inference serving systems Optimize KV‑cache... ...Profile and eliminate performance bottlenecks across GPU, networking, and storage layers Collaborate with Kernels and Research to align execution systems with...PerformanceRelocationVisa sponsorship$170k - $220k
Member of Technical Staff - Infrastructure & LLMs Location: San Francisco, CA (Hybrid) Compensation:... ...strong engineer to join a lean, high-performance team building next-generation inference... ...directly on problems like: Scaling multi-GPU inference workloads Designing...PerformanceFull timeTemporary workImmediate startVisa sponsorshipWork visa$300k
Member of Technical Staff - RL Infrastructure About V max V max is an applied research lab developing... ...and/or RL training. Experience with GPU clusters, distributed training, model... ...observability, testing, debugging, and performance optimization. Ability to work closely...PerformanceWork at officeLocal area- Member of Technical Staff, Infrastructure and Training Systems Location: SF Bay Area or Tokyo, Japan... ...You will work on distributed training, performance optimization, reusable internal... ...the stack affects research velocity: kernel performance, communication overhead,...PerformanceFull time
- ...Design, build, and operate large-scale GPU infrastructure for high-throughput model... ...learning pipelines at scale. Build high-performance inference platforms capable of serving and... ...Improve performance of model execution through kernel-level optimization, model parallelism...PerformanceRelocation package
$180k
...fulfill the need of our high-performance large-scale LLM... ...scale LLMs with JAX (on GPU or TPU) and applying various... ...complex use cases. Kernel Compiler Experience:... ...interview”) during which a member of our team will ask... ...which consists of four technical interviews: # Coding...PerformanceTemporary workRelocation$180k - $300k
...Member Of Technical Staff - Infrastructure Engineer Freiburg (Germany), San Francisco (USA) About... ...optimizing components to extract peak performance from the system (both on application,... ...Python, Bash, Go Kubernetes Nvidia GPU drivers, and operators OTel,...PerformanceWork at officeRemote workWorldwideRelocation2 days per week- ...possible. You will work directly with the technical lead on problems that require deep... ...and directly impacts model performance on real devices. While San Francisco... ...Work Implement and optimize inference kernels for CPU, NPU, and GPU architectures across diverse edge hardware...Performance
- ...scheduling and KV-cache management to support in API Gateway. GPU Kernels Migration to CuTe DSL. Port our in-house CUDA kernels to... ...Python pains and keep up with rapidly growing traffic. Performance Optimization. Profile and fix bottlenecks from network...Performance
- ...techniques and numerical precision trade-offs across different model scales Analyze, profile and debug low-level GPU operations to optimize performance Stay up-to-date on research to bring new ideas to work What We're Looking For Strong grasp of state-...Performance
- ...heterogeneous hardware, enabling step-function improvements in performance and efficiency. Customers deploy through production-grade... ...of AI. About the role Gimlet Labs is seeking a Member of Technical Staff focused on distributed systems. In this role, you will...Performance
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Member of Technical Staff - Kernels & GPU Performance. Be the first to apply!
Related searches
- technical support associate San Francisco, CA
- decision support analyst San Francisco, CA
- desktop support analyst San Francisco, CA
- senior technical analyst San Francisco, CA
- user support analyst San Francisco, CA
- customer support technician San Francisco, CA
- technical support analyst San Francisco, CA
- support analyst San Francisco, CA
- tech assistant San Francisco, CA
- technical support specialist San Francisco, CA

