Principal AI/ML Software Engineer

OCTAVE LLC
Madison, United States of America
yesterday

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Intermediate

Job location

Remote
Madison, United States of America

Tech stack

A/B testing
Artificial Intelligence
Airflow
Amazon Web Services (AWS)
Amazon Web Services (AWS)
Amazon Web Services (AWS)
JIRA
Big Data
Continuous Integration
Database Queries
Monitoring of Systems
Python
Machine Learning
NumPy
Performance Tuning
Power BI
TensorFlow
Prometheus
SQL Databases
Tableau
Datadog
PyTorch
Flask
Large Language Models
Snowflake
Prompt Engineering
Spark
Deep Learning
Generative AI
FastAPI
Pandas
Build Management
Pytest
Scikit Learn
Kubernetes
Information Technology
HuggingFace
Data Analytics
XGBoost
Data Management
Machine Learning Operations
Virtual Agents
Functional Programming
Cloudwatch
Document Classification
GPT
Software Version Control
GXP
Docker
Unsupervised Learning
Databricks

Job description

We are seeking a motivated AI/ML Engineer to build reliable, scalable systems and Generative AI and Agentic AI features, and build and deploy data-driven solutions for our document-based compliance management platform. This role requires a technical expert who can develop, deploy, and maintain ML systems in production environments., Build and deploy Generative AI features using foundation models (AWS Bedrock, OpenAI, Anthropic Claude) and inference pipelines with optimization of latency and cost Design agentic AI systems that autonomously handle compliance workflows, document review, regulatory mapping, and multi-step reasoning tasks Integrate comprehensive LLM evaluation frameworks with development and production systems Build and operate end-to-end MLOps pipelines, deployment systems, monitoring, and rollbacks workflows Implement explainability frameworks (SHAP/LIME) and monitoring dashboards ensuring transparency and regulatory adherence Collaborate with cross-functional teams to translate business needs into ML solutions and communicate insights to stakeholders, LLM evaluation frameworks ensuring 95%+ accuracy for compliance-critical features Prompts for LLMs to achieve specific, high-quality outcomes Agentic AI systems autonomously handling document review and compliance workflows GenAI document understanding features processing millions of regulatory documents Predictive models identifying compliance risks before they occur Real-time semantic search and explainable ML systems meeting regulatory requirements Production MLOps pipelines supporting dozens of models with automated monitoring and retraining

Growth Opportunities Drive adoption of emerging AI technologies and establish best practices Mentor ML engineers Shape AI/ML roadmap and establish center of excellence for compliance AI Collaborate with product leadership on long-term vision for AI-powered compliance

Requirements

Python (5+ years): Production-level experience with Pandas, NumPy, scikit-learn, XGBoost, TensorFlow/PyTorch, Hugging Face Transformers, FastAPI/Flask, MLflow, and pytest SQL: Advanced proficiency with complex queries, window functions, and optimization Machine Learning & NLP: Strong foundation in supervised/unsupervised learning, deep learning, document understanding, text classification, and semantic analysis Generative AI & LLMs: Hands-on experience with foundation models (GPT, Claude, Llama), prompt engineering, RAG architectures, and vector databases (Pinecone, Weaviate, Chroma) MLOps & ModelOps: End-to-end experience with ML pipelines, model versioning, feature stores, drift detection, CI/CD for ML, and Docker containerization LLM Evaluation: Experience with evaluation frameworks (RAGAS, DeepEval), custom metrics, benchmark datasets, and human-in-the-loop validation Cloud & AWS: Experience with AWS services including SageMaker, Bedrock, S3, Lambda, EC2, and CloudWatch Statistics & Experimentation: Strong foundation in statistics, A/B testing, causal inference, and experimental design Visualization: Proficiency with Tableau, Power BI, or Python visualization libraries, 5+ years in data science, ML engineering, or related roles 3+ years building NLP/generative AI applications and implementing MLOps in production Bachelor's or Master's degree in Data Science, Computer Science, Statistics, or related field Track record of deploying ML systems processing large-scale datasets with proper monitoring and governance

Preferred Qualifications Experience with agentic AI frameworks (LangGraph, LangChain, AutoGen, CrewAI) ? Knowledge of Life Sciences/regulated industries (FDA, EMA, ISO, GxP) and compliance management systems Familiarity with big data tools (Spark, Databricks, Snowflake), orchestration (Airflow, Kubeflow), and monitoring tools (Datadog, Prometheus) Experience with LLM fine-tuning, document processing libraries, multi-modal AI, or distributed training Understanding of ML governance, bias detection, model risk management, and data privacy regulations (GDPR, CCPA, HIPAA) Experience working in agile environments with Jira AWS ML certifications or similar credentials

Key Competencies Strong communication skills explaining complex models to technical and nontechnical audiences Ability to work independently and collaboratively in fast-paced environments Proven ability to convert POCs into production-grade solutions Understanding of ethical AI and building trustworthy, explainable systems for regulated environments

About the company

Turning complex operational data into actionable intelligence, Octave connects expertise, real-world conditions and enterprise-scale insight to improve performance, resilience and incident response where it matters most. Octave has more than 7,000 employees in 45 countries. Learn more at octave.com and follow us on LinkedIn. Why work for Octave? All in. Always forward. That's the way we do things around here. We put trust in our people because we believe it's the best way to unleash potential, bring ideas to life, and keep moving ahead. And it's why we're committed to creating an environment that's truly supportive, providing you with the resources you need to support your ambitions, no matter who you are or where you are in the world. Everyone is welcome At Octave, we believe that diverse and inclusive teams are critical to the success of our people and our business. Here, everyone is welcome. As an inclusive workplace, we don't discriminate. In fact, we embrace differences and are fully committed to creating equal opportunities, an inclusive environment, and fairness for all. Respect is the cornerstone of how we operate, so speak up and be yourself. You're valued here. Recruitment Fraud Alert Octave posts all official job opportunities on either or and communicates only from email addresses ending in @octave.com. We never request payment or personal banking information during recruitment. No offers will ever be extended without a proper interview via Teams or in person, never done over email alone. If you suspect fraud, it probably is, and contact us at

Apply for this position