RemoteLeaf Logo
Log in Sign up

Principal Research Scientist – Scaling

3 months 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 216

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

Nebius

Staff / Senior Software Engineer (Agentic Search) – Web Rendering Engineer

Nebius 51-250 Internet Software & Services

Nebius is hiring a Senior Software Engineer to build rendering infrastructure for an agent-native search platform that turns modern web applications into structured content for AI systems.

Poland Czech Republic Estonia Germany Israel Full-time Senior Research Scientist Software Engineer
Angular DOM JavaScript Playwright Puppeteer React Selenium Svelte Vue.js
5 hours, 37 minutes ago
Natera

Senior Bioinformatics Scientist

Natera 1K-5K Pharmaceuticals

Natera is hiring a Senior Bioinformatics Scientist to support early cancer detection by developing computational methods and pipelines for genomic analysis within its R&D bioinformatics team.

United States Full-time Junior Research Scientist
$119k-$148k
AWS Linux NumPy Pandas Python Scikit-learn Shell Scripting
6 hours, 7 minutes ago
Natera

Senior Bioinformatics Scientist

Natera 1K-5K Pharmaceuticals

Natera is hiring a Senior Bioinformatics Scientist to support early cancer detection by developing computational methods and pipelines for genomic analysis within its Research and Development team.

United States Full-time Junior Research Scientist
$131k-$163k
AWS Linux NumPy Pandas Python Scikit-learn Shell Scripting
6 hours, 7 minutes ago
Bitly

Senior Staff Engineer, Discrete Planning (R5440)

Bitly 51-250 Internet Software & Services

Shield AI is seeking a Discrete Planning engineer to develop mission-planning and autonomy software for Hivemind Enterprise in support of intelligent, confidence-inspiring flight behaviors across dynamic multi-vehicle operations.

United States Full-time Lead AI Research Scientist Research Scientist
$230k-$350k
Agile C++
6 hours, 37 minutes 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