ML/AI Engineer

Axiom
Richmond, VA, United States
about 2 months ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Experience required
7 years minimum
Working hours
Regular working hours
Job source

Tech stack

A/B Testing Application Programming Interfaces (APIs) Artificial Intelligence Amazon Web Services Microsoft Azure Continuous Integration Information Engineering Data Warehousing Python (Programming Language) Machine Learning Performance Tuning Search Technologies
+16 more
Software Engineering SQL Databases Data Streaming Supervised Learning Retrieval-Augmented Generation Large Language Models Apache Spark Deep Learning Event Driven Architecture Data Lakes Kubernetes Machine Learning Operations Software Version Control Unsupervised Learning Databricks Microservices

Job description

  • Design, build, train, evaluate, and deploy machine learning models for predictive analytics, classification, NLP, anomaly detection, generative AI, and other applied AI use cases.
  • Develop production-ready AI/ML solutions on a Databricks Lakehouse platform using Python, Spark, MLflow, Delta Lake, and related tools.
  • Build scalable feature pipelines, training workflows, validation processes, and model refresh cycles that are automated, reproducible, and reliable.
  • Own the end-to-end ML lifecycle, including experimentation, model registry, deployment, monitoring, drift detection, model performance tracking, and ongoing optimization.
  • Design modular ML architectures that integrate with APIs, data warehouses, microservices, and downstream applications.
  • Develop LLM-powered applications, prompt engineering strategies, retrieval-augmented generation systems, embeddings, vector search, and related AI capabilities where appropriate.
  • Partner with data engineering, product, platform engineering, and business stakeholders to turn ambiguous business opportunities into measurable AI/ML outcomes.
  • Support experimentation through A/B testing, offline and online evaluation frameworks, statistical validation, and clear communication of results.
  • Document models, systems, decisions, and workflows in a way that enables future engineers and cross-functional teams to adopt and maintain solutions.

Requirements

Are you an experienced ML/AI engineering professional ready to build production-grade intelligent systems?

  • 7+ years of experience in machine learning, applied AI, machine learning engineering, or a similar hands-on technical role.
  • Strong hands-on expertise with Python, Spark, Databricks, MLflow, SQL, and large-scale distributed datasets.
  • Experience building, deploying, and monitoring machine learning models in production environments.
  • Strong understanding of modern ML techniques, including supervised learning, unsupervised learning, deep learning, transformers, embeddings, vector stores, and LLM-based systems.
  • Experience designing reproducible ML pipelines, CI/CD workflows, model deployment patterns, and observability practices.
  • Solid software engineering foundation, including version control, testing, modular architecture, maintainability, and production reliability.
  • Ability to communicate technical concepts clearly to non-technical stakeholders and influence technical direction across teams.
  • Experience working in agile product environments with product managers, engineers, data teams, and business partners.
  • Nice to have: Databricks Model Serving, Unity Catalog, Feature Store, Delta Live Tables, RAG systems, LLM fine-tuning, model distillation, AWS, Azure, Kubernetes, containers, real-time ML, streaming data, or event-driven architectures.
  • Strong curiosity, collaboration skills, ownership mindset, and ability to work through ambiguity with incomplete data or evolving requirements.

Benefits & conditions

Join a mission-driven technology organization focused on improving the aging and long-term care experience for families, caregivers, and care seekers. This team is building modern digital solutions that bring together care options, resources, education, and human support into a more connected and accessible experience. The environment is collaborative, product-focused, and purpose-driven, with a strong emphasis on learning, inclusion, and building technology that creates meaningful real-world impact.

What’s In Store For You:

This is a full-time opportunity available to candidates in Richmond, VA on a hybrid basis or remote candidates residing in approved Eastern Time Zone states. The role offers the opportunity to work on production-grade AI and machine learning solutions, contribute to scalable ML foundations, and partner closely with product, engineering, platform, data, and business stakeholders. Employees are offered comprehensive benefits, retirement savings options, generous paid time off, paid holidays, paid family leave, wellness support, tuition reimbursement, student loan repayment, and training/certification support. This role is not eligible for employment visa sponsorship.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on dice.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:27 min

Managing traffic and tracking costs with Databricks Unity Catalog

Viktoria Semaan Viktoria Semaan · WWC Europe 2026

2:28 min

Understanding Kubernetes architecture and core cluster components

Marc Nimmerrichter · WWC 2022

3:24 min

The governance failures of centralized data lakes

Mario Meir-Huber · LIVE

2:08 min

Essential engineering roles in the generative AI space

Mary Grygleski Mary Grygleski · LIVE

2:50 min

Executing LoRA fine-tuning using serverless Databricks AI runtimes

Viktoria Semaan Viktoria Semaan · WWC Europe 2026

1:31 min

Essential AI and human skills for future teams

Alexander Weißhaupt Alexander Weißhaupt +1 · WWC 2025

Videos

See all

Related articles

See all