Databricks
Databricks is the lakehouse company$1. .$1
Apache Spark • Apache Spark Training • Cloud Computing • Big Data • Data Science
1001 - 5000
GenAI Staff Machine Learning Engineer, Performance Optimization
March 20
🏢 In-office - San Francisco
Databricks
Databricks is the lakehouse company$1. .$1
Apache Spark • Apache Spark Training • Cloud Computing • Big Data • Data Science
1001 - 5000
Description
• Explore and analyze performance bottlenecks in ML training and inference • Design, implement and benchmark libraries and methods to overcome aforementioned bottlenecks • Build tools for performance profiling, analysis, and estimation for ML training and inference • Balance the tradeoff between performance and usability for customers • Facilitate the community through documentation, talks, tutorials, and collaborations • Collaborate with external researchers and leading AI companies on efficiency methods • Curious individuals willing to learn new technologies
Requirements
• Hands on experience with deep learning frameworks like PyTorch, TensorFlow • Experience with high-performance linear algebra libraries such as cuDNN, CUTLASS, Eigen, MKL • General experience with training and deployment of ML models • Experience with compiler technologies relevant to machine learning • Experience with distributed systems development or distributed ML workloads • Hands-on experience with writing CUDA code and knowledge of GPU internals (Preferred) • Publications in top tier ML or System Conferences such as MLSys, ICML, ICLR, KDD, NeurIPS (Preferred)
Benefits
• Annual performance bonus • Equity • $192,000 - $260,000 USD pay range • Eligibility for annual performance bonus • Benefits based on individual factors
Apply Now
