Remote AI Engineer, Databricks

KeyLogic is seeking Databricks Engineers to join our growing team supporting the Department of Energy (DOE)

The position is Remote and requires U.S.

Citizenship

Preference given to candidates with experience supporting the Department of Energy 

Position Requirements:


* U.S.

Citizenship


* PREFERRED Experience with the Department of Energy (DOE)


* Minimum 4 years of engineering experience on Databricks and large-scale big data projects (Delta Lake, DLT, Apache Spark, Unity Catalog).

You've also built API layers that expose data and AI/ML capabilities to enterprise systems.


* Minimum 5 years of experience in Python, Java, or equivalent.

Comfortable with evaluation tooling, logging, monitoring, and observability.


* Minimum 1 year of experience in designing and shipping agentic AI solutions in production.


* Minimum 1 year experience in shipping agentic solutions in production (agents, orchestration, context engineering, RAG, workflows) using AI platforms (OpenAI, Claude, Vertex AI, open source) and Databricks AI tooling (Mosaic AI, Vector Search, Model Serving, MLflow).


* Minimum 2 years of experience in shipping to production: CI/CD, infrastructure as code (Terraform, Helm), monitoring, and debugging.

Practical MLOps and LLMOps experience across the ML and LLM lifecycle (model training, serving, monitoring, prompt management).


* Bachelor's degree 

See Job Description