Applied AI Engineer, Learning Intelligence

About the role

In this role, you will design and maintain a skill and concept graph that connects various educational elements. You will also develop machine learning models to assess learner skill levels and create recommendation systems that guide users through personalized learning paths.

Collaboration is key, as you will work closely with frontend engineers to ensure AI outputs are effectively integrated. You will also define standards for model explainability and partner with product teams to validate the quality of recommendations, ensuring a feedback loop for continuous improvement.

The salary range for this role is between $111,200 and $191,050, depending on various candidate-specific factors.

Skills that matter here

• machine learning: You will apply machine learning techniques to build models that assess learner skills.
• knowledge representation: This role involves designing a skill and concept graph to map relationships in learning.
• product engineering: You will collaborate with various teams to ensure AI features are effectively delivered.
• Python: Advanced Python skills are necessary for developing robust, production-grade applications.
• LLM APIs: Experience with LLM APIs is essential for creating generative features in your projects.
• graph databases: Hands-on experience with graph databases will support the development of the skill graph.

Who this role suits

• You have over five years of experience in applied machine learning or data science.
• You possess strong communication skills and can explain complex concepts to diverse stakeholders.
• You are intellectually curious and enjoy finding straightforward solutions to challenging problems.
• You thrive in collaborative environments and are comfortable partnering with engineers and product teams.

From the employer

What You Will Do

• Design, build, and maintain a skill and concept graph that maps relationships between skills, roles, domains, and learning content.
• Develop ML models that infer learner skill levels from usage patterns, work output, assessments, and profile data (not just self-reported input).
• Build and iterate on recommendation systems that surface the next best module, suggest learning paths, and generate content dynamically.
• Partner with frontend engineers to ensure AI outputs are consumed correctly, surfaced with appropriate context.
• Define explainability standards for model outputs so users and stakeholders understand why a recommendation was made.
• Collaborate with product and content teams to validate recommendation quality and close feedback loops.
• Monitor model performance in production and own the evaluation framework for recommendation quality.

What We Are Looking For

• 5+ years of experience in applied ML or data science, with production recommendation or personalization systems in your background.
• Hands-on experience with knowledge graphs, graph databases, or ontology design.
• Experience with LLM APIs and prompt engineering for generative features.
• Hands-on history of shipping LLM-based systems to production, including large-scale deployment, evaluation frameworks, and agentic workflows.
• Advanced Python proficiency and experience architecting robust, production-grade applications.
• Deep familiarity with the modern AI stack, from retrieval and agent frameworks to complex prompt engineering, model evaluation, and context engineering.
• A high degree of intellectual curiosity and the ability to find elegant, straightforward solutions.
• Exceptional communication skills, with the ability to translate technical logic for varied stakeholders.

Benefits

• Databricks is committed to fair and equitable compensation practices.
• The pay range(s) for this role is listed below and represents the expected base salary range for non-commissionable roles or on-target earnings for commissionable roles.
• Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location.
• The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above.
• For more information regarding which range your location is in visit our page here.
• Zone 1 Pay Range $139,000 — $191,050 USD.
• Zone 2 Pay Range $125,000 — $171,950 USD.
• Zone 3 Pay Range $118,100 — $162,350 USD.
• Zone 4 Pay Range $111,200 — $152,900 USD.