DatabricksMountain View, CAStaff Software Engineer building and scaling Databricks' managed large-scale GPU training platform (AIR). Focus on distributed training performance, scheduling, fault tolerance, and developer experience for thousands of accelerators.
Develops and deploys state-of-the-art GenAI models and systems for Databricks products like Assistant and Genie. Requires 2-8 years ML engineering experience, proficiency in Python/PyTorch/TensorFlow, and expertise in LLMs.
Staff Machine Learning Engineer developing state-of-the-art visual encoders and multimodal models at Pinterest Labs. Prototype visual reasoning tools, train billion-scale models on rich visual-text data, ship to production for recommender systems and VLMs, publish research, and mentor juniors. Requires strong CV/ML background, publications, and PhD or equivalent.
As a Staff Machine Learning Engineer, you will lead the ML strategy and execution for the Advertiser and Seller Experience team, building recommendation systems and context foundations. This role requires deep expertise in recommendation systems and modern agentic AI to shape advertiser and seller workflows.
Designs, implements, and optimizes high-performance GPU kernels for GenAI inference stack. Leads performance improvements, mentors engineers, and collaborates with ML and systems teams. Requires deep kernel programming and GPU architecture expertise.
Leads architecture, development, and optimization of GenAI inference engine for high-throughput, low-latency LLM serving. Requires 6+ years in performance-critical systems, deep ML inference expertise, CUDA/GPU programming, and distributed systems.