Senior Applied AI Engineer

Job title: Senior Applied AI Engineer

About the Role

As a Senior Applied ML/AI Engineer at Databricks, you will apply machine learning and optimization techniques to improve the usability and efficiency of AutoML and other user-facing products, whether through classical models or deep learning approaches. Our team works on a range of challenging problems—from statistical models to deep and foundational models, feature augmentation and auto-tuning—aiming to make Databricks' infrastructure and products more performant and cost-efficient. This is a high-impact role that helps customers get the most value from their data.

What You'll Do

• Build features and run end-to-end systems in a small team of experienced engineers and data scientists.
• Shape the direction of our applied ML investment by engaging with engineering and product teams across the company.
• Drive the development and deployment of state-of-the-art ML/AI models and systems that directly impact the capabilities and performance of Databricks’ products, infrastructure, and services.
• Architect and implement robust, scalable ML infrastructure, including model training and serving components to support seamless integration of AI/ML models into production environments.
• Work on novel modeling techniques in the field of ML for forecasting.
• Possibilities to contribute to the broader AI community by presenting at conferences and actively participating in open source projects, enhancing Databricks’ reputation as an industry leader.

What We're Looking For

• 2-8 years of machine learning engineering experience in high-velocity, high-growth companies
• Strong understanding of both computer systems and statistics
• Experience developing AI/ML systems at scale in production
• Strong track record of ML modeling that goes beyond using standard libraries
• Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews, and deployment
• A large breadth of knowledge or willingness to develop mathematical modelling beyond the ML

Nice to Have

• Experience deploying, scaling, and monitoring models in production
• Understanding of the unique infrastructure challenges posed by training and serving predictions in Tier 0 environments

Compensation & Benefits

• Benefits: At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. Our Commitment to Diversity and Inclusion: We are committed to fostering a diverse and inclusive culture where everyone can excel. Our hiring practices are inclusive and consider applicants without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics.