RemoteLeaf Logo
Log in Sign up

Principal Research Scientist – Scaling

23 hours, 42 minutes 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 146

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

VulnCheck

Sr. Exploit Developer (US)

VulnCheck 11-50 Internet Software & Services

VulnCheck is seeking a Senior Exploit Developer to join its Initial Access Intelligence team, developing exploit and detection artifacts that help customers understand and respond to emerging exploitation activity.

United States Full-time Senior Research Scientist Security Engineer
Cybersecurity Go
2 hours, 44 minutes ago
Hyphen Connect

Embodied AI Researcher (VLA Models)

Hyphen Connect 1-10 staffing & recruiting

An Embodied AI Researcher at an innovative robotics-focused organization will develop and evaluate foundation models that advance robotics through deep learning and zero-shot generalization.

Hong Kong Lead AI Research Scientist Robotics Engineer
Deep Learning Machine Learning Python PyTorch TensorFlow
3 hours, 31 minutes ago
Hyphen Connect

Embodied AI Researcher (VLA Models)

Hyphen Connect 1-10 staffing & recruiting

An Embodied AI Researcher at the company will develop and evaluate foundation models for robotics, advancing deep learning-based systems and zero-shot generalization in practical robotic applications.

United States Lead AI Research Scientist Robotics Engineer
Deep Learning Machine Learning Python PyTorch TensorFlow
3 hours, 31 minutes ago
Hyphen Connect

Embodied AI Researcher (VLA Models)

Hyphen Connect 1-10 staffing & recruiting

An Embodied AI Researcher at an unspecified robotics-focused company will develop and evaluate foundation models for robotic systems, advancing deep learning-based control and zero-shot generalization.

United States Senior AI Research Scientist Robotics Engineer
Deep Learning Machine Learning Python PyTorch TensorFlow
3 hours, 46 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