Software Engineer, ML Platform (New Grad)

About This Role

Databricks is hiring Software Engineers for its ML Platform team to work on MLflow, feature stores, AutoML, and the infrastructure that thousands of data scientists use to build and deploy models. This is an infrastructure and tooling role where your code will be used by the broader ML community.

Job Overview

• Job Title: Software Engineer, ML Platform (New Grad)
• Company: Databricks
• Location: San Francisco, CA / Seattle, WA / New York, NY / Austin, TX
• Job Type: Full-time
• Experience Level: Entry-level (0–2 years)
• Salary Range: $40–$56/hr
• Posted Date: 2026-06-01
• Application Deadline: 2026-07-01

Role Context

Databricks is the company behind the Databricks Lakehouse Platform and the open-source MLflow project. New grad SWEs on the ML Platform team build the tooling that enables end-to-end ML workflows — from experiment tracking and model registry to serving and monitoring. You will write code used by millions of data scientists globally.

Key Responsibilities

• Build new features for MLflow and the Databricks ML Platform in Python and Scala
• Develop APIs and integrations between ML tools and the Databricks Lakehouse
• Implement performance improvements and scalability fixes in ML serving infrastructure
• Write comprehensive tests and contribute to CI/CD pipeline improvements
• Collaborate with ML researchers and users to understand platform pain points
• Contribute to open-source MLflow and present technical work in community forums

Requirements & Skills

• Bachelor's or Master's in CS or Software Engineering (2024–2025 grad)
• Strong Python programming and familiarity with ML workflows (training, evaluation, deployment)
• Understanding of distributed systems or data engineering concepts (Spark is a plus)
• Comfort with REST API design and backend service development
• Open-source contribution experience is a strong differentiator
• Enthusiasm for developer tooling and platform engineering

Benefits & Work Conditions

• $40–$56/hr with Databricks equity and performance bonus
• Hybrid schedule with generous remote flexibility
• Comprehensive health, dental, and vision benefits
• $1,500 annual professional development budget
• Unlimited PTO and generous parental leave

Who Should Apply

New grads who want to build tools that other ML engineers rely on daily. This role is ideal for people who understand ML workflows from a practitioner perspective and want to improve the experience for the broader data science community.

About the Company

Databricks created the Lakehouse architecture and is the primary commercial backer of Apache Spark and MLflow. Its platform is used by over 10,000 organizations globally for large-scale data engineering, analytics, and machine learning workflows.

How to Apply

Apply at databricks.com/careers. The interview includes a coding screen, a systems design discussion, and a behavioral panel. Databricks values clear communication and a genuine interest in the ML tooling space.