Staff Software Engineer - GenAI Performance and Kernel
Posted
$190,900 USD
Tech Stack
Responsibilities
- Lead the design, implementation, benchmarking, and maintenance of core compute kernels optimized for various hardware backends (GPU, accelerators)
- Drive the performance roadmap for kernel-level improvements including vectorization, tensorization, tiling, fusion, mixed precision, sparsity, quantization, memory reuse, scheduling, and auto-tuning
- Integrate kernel optimizations with higher-level ML systems and build/maintain profiling, instrumentation, and verification tooling
- Lead performance investigations and root-cause analysis on inference bottlenecks
- Mentor and guide other engineers on lower-level performance, provide code reviews, and help set best practices
Benefits
- Equity
Culture
Inclusive HiringDiverse LeadershipWork-Life Balance
Requirements
Required: BS/MS/PhD in Computer Science, or a related field
Regions: Us
Get jobs like this in your inbox
Weekly Express, TypeScript hiring trends and salary data — free.
Join 6 engineers getting weekly insights
Get market intelligence in your inbox
Free weekly insights on tech hiring trends, salaries, and in-demand stacks.
Already a subscriber? Sign in
About Databricks
Industry: saas
Size: large
Databricks is a data and AI company that builds and operates the world’s best data and AI infrastructure platform, enabling data teams to turn deep data insights into business impact.
View company profile →Compensation
Base salary: $190,900 USD
Equity: eligibility for equity
Bonus: annual performance bonus