Found.dev
Role Highlights
Languages used
CUDA
Triton
Key skills
ML Research
Computer Science
Code Reviews
Kernel
Optimization
Inference
Machine Learning
Architecture
DataFlow
Infrastructure
PhD
Debugging
AI
Tools, Libraries and Frameworks
GPUs
LOOP
DataBricks
MLFlow
Apache Spark
LLVM
Description
The role involves designing, implementing, and optimizing high-performance GPU kernels for a generative AI inference stack. The engineer will lead the development of low-level compute paths and manage the balance between hardware efficiency and generality. Responsibilities include building profiling and instrumentation tools to monitor performance and correctness. The position requires collaborating with research and product teams to integrate optimizations into production systems. Additionally, the role entails mentoring other engineers and establishing best practices for kernel-level performance engineering.
Required Qualifications and Skills
Candidates must possess a BS, MS, or PhD in Computer Science or a related field. The role requires deep hands-on experience in writing and tuning compute kernels using tools such as CUDA, Triton, or LLVM IR. Applicants need strong knowledge of GPU and accelerator architecture, as well as proficiency in advanced optimization techniques and debugging tools. A proven track record of shipping performance-critical production software is necessary, along with excellent communication and leadership capabilities.
Disclaimer
Disclaimer: Job and company description information and some of the data fields may have been generated via GPT-4 summarisation and could contain inaccuracies. The full external job listing link should always be relied on for authoritative information.
About the company
Databricks
Size
8632
HQ
San Francisco, US
Public/Private
Privately HeldDescription
Databricks is the data and AI company. More than 10,000 organizations worldwide including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark, Delta Lake and MLflow. --- Attention: Databricks applicants Due to reports of phishing, were requesting that all Databricks applicants apply through our official Careers page at databricks.com/company/careers (good news you are here). All official communication from Databricks will come from email addresses ending with @databricks.com or @goodtime.io (our meeting tool).
Related jobs
Sr. Solutions Engineer
Databricks
Apache Spark
Big Data
Data Science
Solution Architect
Zurich, Switzerland
Full Time
Named Enterprise Account Executive
Databricks
Apache Spark
Customer Success
GCP
Data
Stockholm, Sweden
Full Time
Apache Spark
Open Source
GCP
Data
London, UK
Full Time