Machine Learning Ops Engineer Jobs in California

ML Model Deployment & Platform Management

• Lead the design, implementation, and ongoing maintenance of scalable ML infrastructure on Databricks, including ML flow for experiment tracking, model registry, and model serving endpoints.
• Oversee the development of the ML Ops platform and automated pipelines for deploying, monitoring, and maintaining models within production environments.
• Implement robust solutions for model versioning, systematic retraining, and comprehensive artifact management using Databricks Unity Catalog for ML governance.
• Design and manage Databricks Feature Store for consistent feature engineering across training and inference pipelines.

Generative AI & LLM Operations

• Architect and implement Retrieval-Augmented Generation (RAG) systems for document Q&A, enabling business teams to query fund documents, investor letters, and market research.
• Design, deploy, and manage vector database solutions (Databricks Vector Search, Pinecone, or similar) for semantic search and retrieval across enterprise documents.
• Lead LLM fine-tuning and customization initiatives, training models like Claude or open-source alternatives with CIM proprietary data while ensuring data privacy and compliance.
• Develop and optimize document processing pipelines including PDF parsing, chunking strategies, and embedding generation for RAG applications.
• Implement prompt engineering best practices and LLM evaluation frameworks to ensure output quality, relevance, and factual accuracy.
• Build guardrails and safety measures for GenAI applications, including hallucination detection, output validation, and source attribution.

Automation & CI/CD Pipelines

• Design and implement extensive automation across the ML workflow, covering model training, testing, validation, and deployment using Databricks Workflows and Asset Bundles.
• Set up robust CI/CD pipelines for both traditional ML models and GenAI applications, leveraging GitHub Actions, Azure DevOps, or similar tools.
• Automate complex data and model workflows utilizing orchestration tools such as Airflow, Prefect, or Databricks Workflows.

Monitoring, Performance & Reliability

• Implement comprehensive monitoring and alerting systems for real-time tracking of model performance, data quality, and GenAI output quality.
• Utilize specialized tools (Evidently AI, WhyLabs, Prometheus/Grafana) to proactively detect model drift, data quality anomalies, and RAG retrieval degradation.
• Develop evaluation frameworks for GenAI applications including relevance scoring, faithfulness metrics, and human feedback loops.
• Troubleshoot issues within production environments, including debugging model deployment failures, RAG retrieval issues, and LLM response quality problems.

Data & Feature Engineering Support

• Build and maintain sophisticated feature stores on Databricks, ensuring precise alignment between training and inference data pipelines.
• Collaborate with data engineers and information architects to build robust ETL pipelines that feed into the Databricks Lakehouse.
• Design embedding pipelines and vector index management strategies for RAG applications, including incremental updates and versioning.

Security, Compliance & Trustworthy AI

• Integrate robust security measures directly into ML Ops and GenAI pipelines, including access controls via Unity Catalog and data encryption.
• Implement Trustworthy AI guardrails addressing bias detection, explainability, prompt injection prevention, and responsible AI practices.
• Ensure GenAI applications handling sensitive fund and investor data comply with regulatory requirements and internal policies.
• Collaborate with Legal and Compliance to establish AI governance policies and audit trails for model decisions.

Collaboration & Business Partnership

• Engage in extensive collaboration with data scientists, platform engineers, information architects, and DevOps teams to ensure seamless ML/AI integration.
• Partner with business teams (Fund Accounting, FP&A, Investor Relations, Sales, Investments) to identify high-value AI use cases and translate business needs into technical solutions.
• Communicate complex AI concepts in business terms, managing expectations and demonstrating ROI of ML/GenAI initiatives.
• Provide technical mentorship to team members, including refactoring data scientist code for production readiness.

Required:

• Bachelor's or Master's degree in Computer Science, Engineering, Information Systems, or a related field.
• 7+ years of experience as an ML Ops Engineer, ML Engineer, or similar role with production deployment responsibility.
• Expert-level proficiency in Python, complemented by strong skills in Bash scripting.
• Extensive experience designing and implementing cloud solutions on Azure (required) or GCP.
• Deep expertise with Docker and Kubernetes for containerizing and orchestrating ML workloads.
• Hands-on experience with CI/CD tools such as GitHub Actions, Jenkins, GitLab CI, or Azure DevOps.
• Strong SQL proficiency and practical experience with Databricks platform.
• Experience with workflow orchestration tools (Airflow, Prefect, or Databricks Workflows) and monitoring tools (Prometheus, Grafana, Evidently AI).

GenAI/LLM Technical Requirements (NEW):

• Demonstrated experience building and deploying RAG (Retrieval-Augmented Generation) systems in production environments.
• Hands-on experience with vector databases (Databricks Vector Search, Pinecone, Weaviate, Chroma, or Milvus).
• Experience with LLM APIs and frameworks (OpenAI, Anthropic Claude, LangChain, LlamaIndex).
• Understanding of embedding models, chunking strategies, and retrieval optimization techniques.
• Knowledge of prompt engineering best practices and LLM evaluation methodologies.

Databricks Platform Requirements (NEW):

• Experience with ML flow for experiment tracking, model registry, and model serving.
• Familiarity with Databricks Feature Store and Unity Catalog for ML governance.
• Understanding of Delta Lake and Lakehouse architecture for ML data pipelines.
• Experience with Databricks Model Serving endpoints and inference optimization.

Preferred:

• Experience with LLM fine-tuning techniques (LoRA, QLoRA, full fine-tuning) on proprietary data.
• Familiarity with ML frameworks including TensorFlow, PyTorch, Scikit-learn, XGBoost.
• Experience with model serialization (ONNX) and inference optimization.
• Prior experience within financial services, fintech, or private equity sectors.
• Experience building ML/AI infrastructure from scratch in entrepreneurial environments.
• Relevant certifications: Azure AI Engineer Associate, Databricks ML Professional, Google Cloud ML Engineer.

The ideal candidate demonstrates proven experience with model pipeline and registry tools, including the ability to detect and proactively prevent model drift, automate comprehensive model monitoring, and consistently ensure model accuracy. Experience with RAG systems, vector databases, and LLM deployment is essential for this role.

Key Competencies:

• Entrepreneurial Mindset: Comfortable building from scratch in environments without mature processes; thrives with ambiguity and takes initiative.
• Business Partnership: Exceptional ability to translate complex AI/ML concepts for non-technical stakeholders across Fund Accounting, Investor Relations, and Investment teams.
• Cross-Functional Collaboration: Proven track record of building relationships and influence without authority across technical and business functions.
• Business Value Orientation: Focuses on ROI and business outcomes, not just technical elegance; can prioritize initiatives based on business impact.
• Trustworthy AI Champion: Strong awareness of responsible AI practices, including bias mitigation, explainability, security, and compliance requirements.
• Continuous Learner: Stays current with rapidly evolving GenAI landscape; actively experiments with new technologies and approaches.