Data Scientist - Azure Databricks & MLflow (Remote)

Aimlroles; Alexandria (VA); Remote; USD 110,000 - 140,000; Full time

Benefits offered by this job

Medical, dental, and vision

401(k) match up to 4%

Unlimited PTO

Sponsored certifications

Job summary

Peregrine Advisors is seeking a data science/ML engineer to build and productionize AI models for federal clients. You will work on Azure Databricks, MLflow, and Databricks AI tools, advancing retrieval-augmented generation and model serving.

You will join a mission-driven team, own end-to-end model delivery, ensure security and compliance, and grow through sponsored certifications and ongoing training in a fully remote role.

Qualifications

• 4+ years of data science or machine learning engineering experience, including models deployed to production.
• Bachelor's degree in a relevant field.
• Python for data science and ML, incl. pandas, scikit-learn, PySpark.
• Hands-on Azure Databricks work, including MLflow for experiment tracking and model management.
• Deploying and monitoring models in production, including model serving, versioning, and drift monitoring.
• Building LLM applications, including retrieval-augmented generation (RAG).
• Evaluating model and LLM output against defined metrics, test sets, and human review.
• SQL and Git-based version control.
• Delivery experience in a federal government environment, with security and access requirements.
• Proficiency in writing, PowerPoint, and Excel.

Responsibilities

• Develop and train ML models on Azure Databricks.
• Build reusable features in the Databricks Feature Store and serve models through Databricks Model Serving.
• Develop retrieval-augmented AI apps using Agent Bricks and Databricks AI Search.
• Evaluate model quality with MLflow evaluation, including LLM judges and human review.
• Monitor production models for drift and quality; document model reviews.

Skills

Python for ML

Azure Databricks

MLflow

Production deployment

Education

Bachelor's degree

Tools

SQL

Git

Job description

Peregrine Advisors is seeking a data science/ML engineer to build and productionize AI models for federal clients. You will work on Azure Databricks, MLflow, and Databricks AI tools, advancing retrieval-augmented generation and model serving.

You will join a mission-driven team, own end-to-end model delivery, ensure security and compliance, and grow through sponsored certifications and ongoing training in a fully remote role.