Top Machine Learning Jobs in Richmond, VA

The AI/Machine Learning Engineer will develop systems for automating health benefits decisions, integrating large language models with rule-driven patterns, and ensuring compliance with regulations.

The Machine Learning Engineer will develop software solutions, optimize data pipelines, and deploy machine learning models using cloud platforms and best practices.

Manage and mentor a team of machine learning and data engineers, overseeing ML model development, data pipelines, and collaborating with product teams on technical requirements.

Drive delivery excellence and customer success by leading the architecture and implementation of AI/ML solutions while ensuring account growth and client engagement.

Design, build, and deploy K-12 AI features (lesson planning, grading, tutoring, admin analytics) while ensuring FERPA/COPPA compliance and child-safety guardrails. Measure impact, iterate with curriculum and legal partners, and maintain production-grade documentation, testing, and runbooks for deployed AI systems.

Develop prototypes, PoCs, and MVPs for GenAI solutions. Apply deep learning and ML principles using Python, Hugging Face, LangChain, OpenAI API, TensorFlow/Keras/PyTorch, and cloud model services. Work with multi-modal data and intelligent agent tools. Be self-motivated, collaborative, and focused on solving hard AI/GenAI problems.

Lead development and productionization of large-scale ranking and personalization ML systems. Build data pipelines, features, and models (batch and real-time), collaborate cross-functionally, run experiments, and improve infrastructure for search, recommendation, and marketplace optimization.

As a Machine Learning Engineer at Shipwell, you'll build and scale AI-powered logistics solutions, maintaining data pipelines and ML infrastructure across the organization.

You will build and own end-to-end ML systems, focusing on training, evaluation, deployment, and integration with applications, while ensuring quality and efficiency.

The Senior Machine Learning Engineer will build and own production ML systems, manage end-to-end workflow, debug issues, and mentor others.

Design, build, test, deploy, and maintain production AI capabilities (agents, workflows, models, MCP servers, APIs) across a SaaS dental platform. Train and operate ML models, implement RAG and vector DB solutions, use AI-assisted tools, collaborate in design and code reviews, and support production reliability and monitoring.

Design, optimize, and operate production LLM inference and training pipelines. Improve latency, throughput, and GPU utilization using techniques like batching, quantization, FlashAttention, and kernel-level profiling. Deploy and autoscale multiple models on shared GKE GPU clusters, consult with clients on performance and cost requirements, and carry prototypes to robust, scalable production services.

Design, develop, and deploy end-to-end AI/ML applications (RAG, chatbots, agentic workflows). Build scalable data pipelines, apply graph analytics, entity resolution, and NLP for risk scoring, anomaly detection, and investigative prioritization. Develop models, REST APIs, and deploy/manage solutions in AWS to support federal mission use cases.

Build and re-architect Shipt's LLM-native search stack including retrieval and ranking systems. Design multi-task and multi-objective models, integrate personalization signals, develop hybrid dense/sparse retrieval, and apply generative retrieval and query expansion. Analyze search data, report findings to leadership, and partner with senior scientists and ML engineers to improve relevance, revenue trade-offs, and operational metrics.

Develop and scale reinforcement-learning systems that close the loop between automated labs and reward models. Build synthetic data engines and inference infrastructure for large-scale training, maintain evaluations of learned policies using proprietary experimental datasets, and address real-world challenges like experimental noise and expensive queries.

As a Senior Machine Learning Engineer, you'll design and develop infrastructure and models for AI-powered applications while collaborating with teams to enhance product functionality.

As a Senior Staff Machine Learning Engineer, you will lead the development and deployment of machine learning models, ensuring they drive business growth and meet performance requirements. You will collaborate with teams, manage data pipelines, and implement state-of-the-art solutions while continuously improving model performance.

The role involves developing and deploying machine learning models for forecasting within the Camus platform, requiring collaboration and technical depth in ML.

Design and develop AI capabilities and scalable microservices. Collaborate to integrate AI models and ensure performance in cloud environments.

Lead design, training, fine-tuning, evaluation, and production deployment of TTS/voice-generation models. Translate research into reliable, low-latency production systems, define ML best practices, and collaborate across research, engineering, and product to drive execution.

As an MLOps/ML Platform Engineer, you will build and manage ML systems, optimize workloads, and ensure production model reliability while collaborating across teams.

Seeking a Senior Machine Learning Engineer to design and implement ML systems, optimize deployment, and collaborate with data teams, focusing on scalability and performance.

The Director of Data Engineering, ML & AI leads data strategy and AI capabilities, managing a team to enhance data infrastructure, machine learning systems, and sustainability analytics while driving company-wide AI initiatives.

Manage AI/ML projects from Proof of Concept to Minimum Viable Product using Agile methodologies while ensuring stakeholder communication and project alignment.

The R&D Engineer will build and refine software solutions for defense use cases, prototype systems, and convert stakeholder feedback into technical requirements. Collaboration with engineering teams and customer-oriented engagements are key responsibilities.