Software Engineer Fullstack (AI/ML)

TL;DR

Software Engineer Fullstack (AI/ML): Building and maintaining open-source AI/ML platforms to enable users to train, deploy, and monitor models and GenAI agents at scale with an accent on collaboration with the larger AI/ML community to contribute to open-source projects and advance AIOps capabilities. Focus on design and implementation of platform integrations with various frameworks in the AI/ML ecosystem.

Location: Tokyo, Japan

Company

Databricks is the data and AI company that provides Data Intelligence Platform to unify and democratize data, analytics and AI.

What you will do

• Design and implement platform capabilities to support the AI/ML development and productionization lifecycle.
• 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.
• Ensure the latest AI/ML tooling advancements are available to Databricks’ customers.
• Mentor and guide junior engineers on the team by helping with project planning, technical decisions, and code and document review.

Requirements

• BS (or higher) in Computer Science, or a related field.
• 3+ 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.
• Familiarity with AI/ML and AIOps concepts and technologies, such as model training, deployment, and monitoring.

Nice to have

• Deep understanding and experience in working with agent frameworks such as LangChain, LlamaIndex, DSPy, or other similar projects.
• Significant contributions to open-source projects in the AI/ML domain, such as SparkML, TensorFlow, PyTorch, MLflow, or other similar projects.

Culture & Benefits

• Comprehensive benefits and perks that meet the needs of all of our employees.
• Fostering a diverse and inclusive culture where everyone can excel.