RemoteLeaf Logo
Log in Sign up

Principal Research Scientist – Scaling

1 month, 1 week ago
United States
Full-time
Lead
AI Research Scientist
Software Development
Apache Spark Deep Learning Generative AI LLM Machine Learning MLflow Python PyTorch Reinforcement Learning
Apply Now
Databricks

Databricks

Databricks is the pioneering data intelligence platform, empowering organizations worldwide to solve complex data challenges with AI-driven analytics solutions.

IT Services
1K-5K
Founded 2013
$4450M raised
View All Jobs 209

Description

  • Lead and grow a multidisciplinary research team focused on LLM scaling, efficiency, and systems performance.
  • Define and execute the scaling research roadmap in alignment with Databricks’ strategic objectives.
  • Drive algorithmic innovations for large-scale training and inference, including optimizers, low-precision techniques, and model adaptation methods.
  • Oversee the design and execution of large-scale experiments and benchmark results against state-of-the-art methods.
  • Optimize distributed training, parallelism, memory management, and hardware utilization in collaboration with systems and infrastructure teams.
  • Translate research breakthroughs into customer-facing capabilities in the Databricks AI platform.
  • Establish metrics, evaluation protocols, and best practices for scaling-focused research and drive adoption across the organization.
  • Champion responsible deployment by ensuring model behavior, reliability, and safety remain first-class considerations.
  • Work hands-on with the team to develop high-quality Python and PyTorch code for research, prototyping, and production integration.
  • Mentor and develop research scientists and engineers through technical guidance and career support.

Requirements

  • Proven ability to lead a research team developing novel techniques for foundation model efficiency or related topics.
  • Strong track record of industry impact.
  • Deep expertise in at least one of: generative AI, LLMs, distributed ML systems, model optimization, or responsible AI.
  • Strong emphasis on scaling and efficiency for large-scale neural networks.
  • Strong programming skills and demonstrated ability to write high-quality, efficient code in Python and PyTorch.
  • Demonstrated ability to translate research innovation into scalable product capabilities with product and engineering teams.
  • Excellent communication, leadership, and stakeholder management skills.
  • Experience influencing cross-functional roadmaps and aligning research with business impact.
  • Prior work at the intersection of systems and ML, such as distributed training frameworks, compiler and kernel optimization, or memory-/compute-efficient model design (preferred).
  • Strong industry and academic network in large-scale ML, with ongoing collaborations or conference service such as PC or area chair roles (preferred).
  • First-author publications at top ML/systems conferences such as ICLR, ICML, NeurIPS, or MLSys, or influential open-source contributions / widely used deployed systems, especially in optimization or efficiency (preferred).

Benefits

  • Competitive base salary range of $280,000 to $350,000 USD.
  • Eligibility for an annual performance bonus.
  • Eligibility for equity as part of the total compensation package.
  • Comprehensive benefits and perks offered regionally.
  • Compensation may be adjusted based on skills, experience, certifications, training, and work location.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Elastic

Principal Data Scientist - Agent Builder

Elastic 1K-5K Internet Software & Services

Elastic is hiring a Principal Data Scientist to lead evaluation and quality improvements for its conversational and agentic search platform built on Elasticsearch.

United Kingdom Lead AI Research Scientist Data Scientist
Elasticsearch LLM Machine Learning NLP Pandas Python PyTorch Transformers
1 day, 1 hour ago
Natera

Senior Machine Learning Scientist, Agentic AI

Natera 1K-5K Pharmaceuticals

Natera is hiring a Senior Machine Learning Scientist to lead agentic AI and multi-modal foundation model work for oncology-focused R&D that turns complex molecular and clinical data into verifiable insights for biomarker and therapeutic discovery.

United States Full-time Senior AI Research Scientist
$160k-$212k
LLM Machine Learning PyTorch React Statistics
1 day, 7 hours ago
Toptal

AI Researcher

Toptal 5K-10K Construction & Engineering

Toptal is hiring a remote AI Researcher to advance agentic AI systems using proprietary real-world interaction data, with a focus on improving reasoning, multimodal learning, and production-ready model behavior.

Asia Europe Full-time Senior AI Research Scientist
Hugging Face LLM Machine Learning Python PyTorch Reinforcement Learning
2 days, 2 hours ago
The Trevor Project

Research Scientist

The Trevor Project 251-1K Diversified Consumer Services

The Trevor Project is hiring a remote Research Scientist to lead quantitative research on LGBTQ+ youth mental health and suicidology, contributing to surveys, publications, and organization-wide data-informed initiatives.

United States Full-time Senior Research Scientist
$75k-$91k
3 days, 17 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers