← All jobs · Databricks

Senior Applied AI Engineer

Databricks ·
75
AI-Agency
B82 U65
📍 Belgrade, RS Senior 2–8+ yrs
PythonPyTorchTensorFlowApache SparkMLflowDelta Lake
TL;DR

Senior Applied AI Engineer at Databricks building ML/AI features for AutoML and user-facing products. Focus on classification, regression, forecasting, and recommendations at scale in production environments.

Apply at Databricks →
you'll be redirected to the company's career page

Job description

P-1439

 

As a Senior Applied ML/AI Engineer at Databricks, you will apply machine learning and optimization algorithms to improve the usability and efficiency of the current AutoML and several other user-facing products that will benefit from better classification, regression, forecasting, and recommendations, either classical or based on deep learning. From statistical models all the way down to deep and foundational models, feature augmentation and auto-tuning, our Applied ML/AI team works on some of the most complex, most interesting problems facing businesses, making Databricks' infrastructure and products as performant and cost-efficient as possible. This is a high-impact problem as our customers look at us to deliver the most out of their data.

The impact you will have:

What we look for:

Why Join Us?

At Databricks, we are building state-of-the-art AI solutions that redefine how users interact with data and our products. You’ll have the opportunity to shape the future of AI-driven products at Databricks, work with cutting-edge models, and collaborate with a world-class team of AI and ML experts.

If you're excited about pushing the boundaries of AI in real-world applications, we’d love to hear from you!

About Databricks

Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on TwitterLinkedIn and Facebook.

Benefits

At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here.

Our Commitment to Diversity and Inclusion

At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered 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.

Compliance

If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

Apply at Databricks →

More open roles at Databricks

Databricks ·
Software Engineer - GenAI inference
📍 San Francisco, US 💰 $142K–$204K · Mid
Software engineer at Databricks building the GenAI inference engine for the Foundation Model API. Focus on LLM serving systems optimization, GPU kernels, distributed inference infrastructure, and performance bottleneck analysis.
CUDAcuBLAScuDNNNCCLPythonPyTorch
81
AI-core
Databricks ·
Staff Software Engineer - GenAI inference
📍 San Francisco, US 💰 $190K–$232K · Staff
Staff software engineer at Databricks building the GenAI inference engine for the Foundation Model API. Focus on architecture, optimization, and scaling of LLM serving across GPUs and accelerators.
CUDAcuBLAScuDNNNCCLPyTorchMLflow
81
AI-core
Databricks ·
Senior GenAI Research Engineer - Optimization and Kernels
📍 San Francisco, US 💰 $166K–$225K · Senior
Senior GenAI Research Engineer at Databricks optimizing GPU kernels and distributed training for large language models. Focus on CUDA kernel development, parallelism strategies, and performance optimization for LLM training workloads.
CUDAPyTorchNVIDIA GPUPythonDeepSpeedMegatron-LM
81
AI-core
Databricks ·
PhD GenAI Research Scientist Intern
📍 San Francisco, US 💰 $112K–$124K · Intern
PhD GenAI Research Scientist Intern at Databricks working on domain adaptation for LLMs and AI systems. Focus on fine-tuning, evaluation, retrieval augmentation, and efficient inference for enterprise applications.
PyTorch
81
AI-core
Databricks ·
Sr. Developer Advocate, Databricks AI Agentic Systems
📍 San Francisco, US 🛠 AI tools welcome at work · Senior
Senior Developer Advocate at Databricks focused on AI Agentic Systems and MLOps. Drive adoption of Agent Bricks through reference implementations, technical content, community building, and product advocacy for enterprise AI governance.
PythonPyTorchscikit-learnLangChainLlamaIndexDSPy
80
AI-core