About this role
Develop, deploy, and operationalize machine learning models including LLMs, RAGs, and AI agent systems using Databricks, MLflow, and AWS; collaborate with data engineers and stakeholders to align ML pipelines with a Medallion Architecture and maintain production model performance. Responsibilities include building training/inference pipelines, implementing MLOps/CI-CD workflows, monitoring models for drift and latency, and mentoring junior team members.
Skills for this role
Pythonscikit-learnXGBoostPyTorchTensorFlowHugging FaceDatabricksMLflowPySparkMLOpsCI/CDAWS S3AWS EC2AWS LambdaAWS SageMakerAWS Step FunctionsModel deploymentModel monitoringDrift detectionProductionizing modelsFeature engineeringMedallion ArchitectureTraining pipelinesInference pipelinesLLMsRAGsAI agent architecturesStreamlitGradioSoftware engineering