Description
As a Sr. Software Engineer on the AI OSS Ecosystem Team, you will play a key role in building and maintaining our open-source AI/ML platforms to enable users to train, deploy and monitor models and GenAI agents at scale.
Your responsibilities will include designing and implementing platform capabilities to support the AI/ML development and productionization lifecycle, including training, evaluation, deployment, monitoring, and management of models and agents.
You will also design and implement platform integrations with various frameworks in the AI/ML ecosystem, collaborate with the AI/ML community across the world to advance the state-of-the-art in AIOps, and ensure the latest AI/ML tooling advancements are available to Databricks' customers.
Additionally, you will mentor and guide junior engineers on the team by helping with project planning, technical decisions, and code and document review.
We are looking for a highly skilled and experienced software engineer with a strong background in AI/ML and a passion for building and maintaining open-source platforms.
The ideal candidate will have a BS (or higher) in Computer Science, or a related field, and 5+ years of hands-on experience in building production systems using at least one of the following programming languages: Python (Preferred), Scala and Java.
Experience building and maintaining software tools and frameworks for AI/ML, ideally in an open-source environment, is also required.
Familiarity with AI/ML and AIOps concepts and technologies, such as model training, deployment, and monitoring, is essential.
A deep understanding and experience in working with agent frameworks such as LangChain, LlamaIndex, DSPy, or other similar projects is preferred.
Significant contributions to open-source projects in the AI/ML domain, such as SparkML, TensorFlow, PyTorch, MLflow, or other similar projects, are also preferred.
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, please click here.
We are committed to diversity and inclusion and welcome applications from candidates of all backgrounds.