Senior AI/ML Engineer – Kernel Optimization
PrismML· Engineering
Hybrid (Pasadena;San Francisco), RemoteFull-timePosted 3mo ago
About the role
Senior-level AI/ML engineer focused on systems and kernel development to optimize low-level performance across the model stack, including 1-bit and compressed representations, across diverse hardware platforms.
Responsibilities
- Designing and implementing custom kernels for model execution across GPUs and other accelerator hardware
- Optimizing inference performance for highly efficient model representations (e.g., 1-bit or quantized models)
- Improving throughput, latency, and memory efficiency across different inference runtimes and deployment environments
- Collaborating with model and systems teams to co-design architectures and execution strategies for maximum performance
- Profiling and debugging performance bottlenecks at the kernel, runtime, and system levels
- Translating advances in hardware-aware optimization into production-ready systems
- Providing technical leadership through mentoring, design reviews, and cross-functional collaboration
Qualifications
- 5–8+ years of experience in systems engineering, machine learning infrastructure, or related fields
- Strong programming skills in C/C++ and/or CUDA (or equivalent low-level languages) with production-quality code
- Hands-on experience developing and optimizing kernels for GPUs or other accelerators
- Solid understanding of computer architecture, memory hierarchies, and parallel programming models
- Experience profiling and optimizing performance-critical systems
- Proven ability to mentor and lead other engineers
Skills
Required:C/C++CUDAKernel optimizationPerformance profiling
Nice to have:Hardware-aware optimization
Last updated Oct 22, 2026. Interested? Apply on the company site.
About PrismML
Concentrating intelligence
PrismML — Concentrating intelligence
Bonsai 27B Bonsai 8B Bonsai 4B Bonsai 1.7B Bonsai Image
More open roles at PrismML
- Senior AI/ML Engineer – Forward Deployment2mo agoHybrid (Pasadena; San Francisco) · PrismML Careers / Engineering
- Staff AI/ML Engineer – Large-Scale Systems3mo agoHybrid (Pasadena or San Francisco)
- Senior AI/ML Engineer – Post-Training Platform3mo agoHybrid (Pasadena or San Francisco)